BZOJ2743: [HEOI2012]采花

Description

萧芸斓是Z国的公主,平时的一大爱好是采花。
今天天气晴朗,阳光明媚,公主清晨便去了皇宫中新建的花园采花。花园足够大,容纳了n朵花,花有c种颜色(用整数1-c表示),且花是排成一排的,以便于公主采花。公主每次采花后会统计采到的花的颜色数,颜色数越多她会越高兴!同时,她有一癖好,她不允许最后自己采到的花中,某一颜色的花只有一朵。为此,公主每采一朵花,要么此前已采到此颜色的花,要么有相当正确的直觉告诉她,她必能再次采到此颜色的花。由于时间关系,公主只能走过花园连续的一段进行采花,便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了m个行程,然后一一向你询问公主能采到多少朵花(她知道你是编程高手,定能快速给出答案!),最后会选择令公主最高兴的行程(为了拿到更多奖金!)。

Input

 第一行四个空格隔开的整数n、c以及m。接下来一行n个空格隔开的整数,每个数在[1, c]间,第i个数表示第i朵花的颜色。接下来m行每行两个空格隔开的整数l和r(l ≤ r),表示女仆安排的行程为公主经过第l到第r朵花进行采花。

Output

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

Sample Input

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

Sample Output

2
0 0 1 0
【样例说明】
询问[1, 5]:公主采颜色为1和2的花,由于颜色3的花只有一朵,公主不采;询问[1, 2]:颜色1和颜色2的花均只有一朵,公主不采;
询问[2, 2]:颜色2的花只有一朵,公主不采;
询问[2, 3]:由于颜色2的花有两朵,公主采颜色2的花;
询问[3, 5]:颜色1、2、3的花各一朵,公主不采。
 
第一眼看错题了,WA+1
然后偷懒写个莫队,TLE+2
乖乖码正解:
离线,按左端点排序,那么对于每种颜色只有第二次出现是有意义的。
所以当L-1移出边界后,要标记从[L,next[L-1]]的答案-1,差分后用树状数组做一下就行了。
#include<cstdio>
#include<cctype>
#include<queue>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
    if(head==tail) {
        int l=fread(buffer,1,BufferSize,stdin);
        tail=(head=buffer)+l;
    }
    return *head++;
}
inline int read() {
    int x=0,f=1;char c=Getchar();
    for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=1000010;
int n,m,r,next[maxn],last[maxn],A[maxn],ans[maxn];
struct Query {
    int l,r,id;
    bool operator < (const Query& ths) const {
        return l<ths.l;
    }
}Q[maxn];
int c[maxn];
int sum(int x) {
    int res=0;
    for(;x;x-=x&-x) res+=c[x];
    return res;
}
void add(int x,int v) {
    if(!x) return;
    for(;x<=n;x+=x&-x) c[x]+=v;
}
int main() {
    n=read();r=read();m=read();
    rep(i,1,n) A[i]=read();
    rep(i,1,m) Q[Q[i].id=i].l=read(),Q[i].r=read();
    dwn(i,n,1) next[i]=last[A[i]],last[A[i]]=i;
    rep(i,1,r) add(next[last[i]],1);
    int l=1;
    sort(Q+1,Q+m+1);
    rep(i,1,m) {
        while(l<Q[i].l) {
            add(next[l],-1);add(next[next[l]],1);
            l++;
        }
        ans[Q[i].id]=sum(Q[i].r)-sum(Q[i].l-1);
    }
    rep(i,1,m) printf("%d
",ans[i]);
    return 0;
}
View Code
原文地址:https://www.cnblogs.com/wzj-is-a-juruo/p/4995915.html