BZOJ 2743 【HEOI2012】 采花

时间:2021-03-10 06:45:24

题目链接:采花

  这道题一眼看去,一个很显然的想法就是莫队。但是数据范围是\(10^6\)级别的,莫队显然已经过不去了。

  其实感觉这道题和以前写过的一道题HH的项链很像。只不过那道题要求的是区间出现次数至少一次的元素个数,这道题次数变成了两次。然而实际上并没有什么不同。

  我们考虑将询问按照右端点排序。这样的话,我们只需要从左到右扫一遍区间,然后一次处理询问即可。如果询问的是至少出现一次的元素个数的话,那么可以先预处理出位置$i$的元素上一次出现的位置$pre_i$,那么我们每扫到一个点$i$的时候就可以把这个点的权值加$1$,然后把$pre_i$权值减$1$。树状数组维护,询问时查询区间和即可。这样实际上就是把每个元素在当前扫过的位置中最靠右的位置赋值为$1$,其它的位置赋值为$0$。显然像这样尽量靠右选取元素最优,因为这样取消了区间右端点的限制,只需要考虑左端点的限制。

  这道题询问的是出现至少两次的元素个数,那么我们就稍微变一下。我们扫到一个位置$i$的元素时将$pre_i$权值加$1$,然后再将$pre_{pre_i}$权值减$1$即可。实际上和上面的方法是一样的。

  下面贴代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
#define maxn 1000010 using namespace std;
typedef long long llg; struct data{
int l,r,b;
}s[maxn];
int n,C,m,a[maxn],ans[maxn];
int pr[maxn],c[maxn]; int getint(){
int w=0;bool q=0;
char c=getchar();
while((c>'9'||c<'0')&&c!='-') c=getchar();
if(c=='-') c=getchar(),q=1;
while(c>='0'&&c<='9') w=w*10+c-'0',c=getchar();
return q?-w:w;
} bool cmp(data a,data b){return a.r<b.r;}
void add(int x,int y){while(x<=n) c[x]+=y,x+=x&(-x);}
int sum(int x){
int t=0;
while(x) t+=c[x],x-=x&(-x);
return t;
} int main(){
File("a");
n=getint(); C=getint(); m=getint();
for(int i=1;i<=n;i++){
a[i]=getint();
pr[i]=c[a[i]]; c[a[i]]=i;
}
for(int i=1;i<=C;i++) c[i]=0;
for(int i=1;i<=m;i++) s[i].l=getint(),s[i].r=getint(),s[i].b=i;
sort(s+1,s+m+1,cmp);
for(int i=1,j=1;i<=n;i++){
if(pr[i]) add(pr[i],1); if(pr[pr[i]]) add(pr[pr[i]],-1);
while(j<=m && i==s[j].r) ans[s[j].b]=sum(s[j].r)-sum(s[j].l-1),j++;
}
for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
return 0;
}