Description
Input
Output
Sample Input
1 2 2 3 1
1 5
1 2
2 2
2 3
3 5
Sample Output
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的项链加强版~
还是按照HH的项链里的思想,先预处理出上一个与i颜色相同的点pre[i],将询问按右端点排序,然后一边往树状数组里扔点一边处理询问,区别是HH的项链里我们是updata(pre[i],-1),updata(i,1),而这里改成updata(pre[pre[i]],-1),updata(pre[i],1)就好了~
#include <cstdio> #include <iostream> #include <algorithm> #include <cstring> using namespace std; int n,C,m,tot; const int maxn=1000010; int pre[maxn],s[maxn],p[maxn],head[maxn],ans[maxn]; struct QUERY { int qa,qb,org; }q[maxn]; bool cmp(QUERY a,QUERY b) { return a.qb<b.qb; } void updata(int x,int v) { if(!x) return ; tot+=v; for(int i=x;i<=n;i+=i&-i) s[i]+=v; } int query(int x) { int i,ret=0; for(i=x;i;i-=i&-i) ret+=s[i]; return ret; } int main() { scanf("%d%d%d",&n,&C,&m); int i,j; for(i=1;i<=n;i++) { scanf("%d",&p[i]); pre[i]=head[p[i]]; head[p[i]]=i; } for(i=1;i<=m;i++) scanf("%d%d",&q[i].qa,&q[i].qb),q[i].org=i; sort(q+1,q+m+1,cmp); for(i=j=1;i<=m;i++) { for(;j<=q[i].qb;j++) updata(pre[pre[j]],-1),updata(pre[j],1); ans[q[i].org]=tot-query(q[i].qa-1); } for(i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }
| 欢迎来原网站坐坐! >原文链接<