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
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的花各一朵,公主不采。
第一眼看错题了,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\n",ans[i]); return 0; }