洛谷P2056 采花

题目描述

萧芸斓是 Z国的公主，平时的一大爱好是采花。

今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。花园足够大，容纳了 n 朵花，花有 c 种颜色（用整数 1-c 表示） ，且花是排成一排的，以便于公主采花。

公主每次采花后会统计采到的花的颜色数， 颜色数越多她会越高兴！ 同时， 她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。 由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了 m 个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！ ） ，最后会选择令公主最高兴的行程（为了拿到更多奖金！ ） 。

输入输出格式

输入格式：

第一行四个空格隔开的整数 n、c 以及 m。

接下来一行 n 个空格隔开的整数，每个数在[1, c]间，第i 个数表示第 i 朵花的颜色。

接下来 m 行每行两个空格隔开的整数 l 和 r（l ≤ r） ，表示女仆安排的行程为公主经过第 l 到第r 朵花进行采花。

输出格式：

共m行， 每行一个整数， 第i个数表示公主在女仆的第i个行程中能采到的花的颜色数。

输入输出样例

输入样例#1： 

5 3 5 
1 2 2 3 1
1 5 
1 2 
2 2 
2 3 
3 5

输出样例#1： 

2 
0 
0 
1 
0

说明

对于100%的数据，1 ≤ n ≤10^5，c ≤ n，m ≤ 10^5。

好久没有乱搞过莫队了。。。

莫队直接上模板。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#define MAXN 100010
using namespace std;
int n,m,c,num[MAXN],sum[MAXN],ans[MAXN];
struct node{
       int l,r,f;
}que[MAXN];
inline int read(){
       int date=0,w=1;char c=0;
       while(c!='-'&&(c<'0'||c>'9'))c=getchar();
       if(c=='-'){w=-1;c=getchar();}
       while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
       return date*w;
}
int cmp1(const node &a,const node &b){
    return a.l<b.l;
}
int cmp2(const node &a,const node &b){
    if(a.r==b.r)
    return a.l<b.l;
    return a.r<b.r;
}
int main(){
    int nowi=1,nowj=0,x,d,left,right,s=0;
    n=read();c=read();m=read();
    for(int i=1;i<=n;i++)
    num[i]=read();
    for(int i=1;i<=m;i++){
            que[i].l=read();que[i].r=read();
            que[i].f=i;
            }
    sort(que+1,que+m+1,cmp1);
    x=(int)sqrt((double)n);
    while(nowi<=m){
                   nowj++;
                   d=nowi;
                   while(que[nowi].l<=nowj*x&&nowi<=m)
                   nowi++;
                   sort(que+d,que+nowi,cmp2);
                   if(nowj==x){
                               sort(que+d,que+m+1,cmp2);
                               break;
                               }
                   }
    left=1;
    right=0;
    for(int i=1;i<=m;i++){
            if(que[i].l<left)
            for(int j=que[i].l;j<left;j++){
                    if(sum[num[j]]==1)
                    s++;
                    sum[num[j]]++;
                    }
            else
            for(int j=left;j<que[i].l;j++){
                    sum[num[j]]--;
                    if(sum[num[j]]==1)
                    s--;
                    }
            left=que[i].l;
            if(que[i].r>right)
            for(int j=right+1;j<=que[i].r;j++){
                    if(sum[num[j]]==1)
                    s++;
                    sum[num[j]]++;
                    }
            else
            for(int j=que[i].r+1;j<=right;j++){
                    sum[num[j]]--;
                    if(sum[num[j]]==1)
                    s--;
                    }
            right=que[i].r;
            ans[que[i].f]=s;
            }
    for(int i=1;i<=m;i++)
    printf("%d\n",ans[i]);
    return 0;
}