BZOJ2743: [HEOI2012]采花
2743: [HEOI2012]采花
Time Limit: 15 Sec Memory Limit: 128 MBSubmit: 2386 Solved: 1231
[Submit][Status][Discuss]
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
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的花各一朵,公主不采。
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的花各一朵,公主不采。
HINT
【数据范围】
对于100%的数据,1 ≤ n ≤ 10^6,c ≤ n,m ≤10^6。
思路{
如果数据范围小一点的话直接莫队就可以了.
想到了HH的项链.......
那么首先按照L和Rsort询问把所有颜色的第一个数的nxt出现位置+1.在往后扫的过程中nxt[i]位置+1,nxt[nxt[i]]位置-1,处理到l.
再树状数组统计即可,
画一下图模拟一下发现很明显这是对的.
}
#include<bits/stdc++.h> #define RG register #define il inline #define db double #define LL long long #define N 1000010 #define lowbit (i&-i) using namespace std; int tree[N],n,nxt[N],beg[N],col[N],con[N],m,c;bool vis[N]; void Insert(int pos,int num){for(int i=pos;i<=n;i+=lowbit)tree[i]+=num;return;} int Query(int pos){int sum(0);for(int i=pos;i;i-=lowbit)sum+=tree[i];return sum;} struct q{ int l,r,id,ans; void read(int x){id=x;scanf("%d%d",&l,&r);} }Q[N]; bool comp(const q & a,const q & b){return a.l==b.l?a.r<b.r:a.l<b.l;} bool Comp(const q & a,const q & b){return a.id<b.id;} int main(){ scanf("%d%d%d",&n,&c,&m); for(int i=1;i<=n;++i)scanf("%d",&col[i]),beg[col[i]]=con[col[i]]=i; for(int i=n;i;--i) if(beg[col[i]]!=i){ nxt[i]=con[col[i]]; con[col[i]]=i; } for(int i=1;i<=c;++i)if(nxt[con[i]])Insert(nxt[con[i]],1); for(int i=1;i<=m;++i)Q[i].read(i); sort(Q+1,Q+m+1,comp); int l=1; for(int i=1;i<=m;++i){ while(l<Q[i].l){ if(nxt[l])Insert(nxt[l],-1); if(nxt[nxt[l]])Insert(nxt[nxt[l]],1); l++; } Q[i].ans=Query(Q[i].r)-Query(Q[i].l-1); } sort(Q+1,Q+m+1,Comp); for(int i=1;i<=m;++i)cout<<Q[i].ans<<"\n"; return 0; }