您的位置:首页 > 其它

bzoj 2743: [HEOI2012]采花【树状数组】

2018-09-07 09:21 393 查看

离线,按照l排序
注意到在区间里出现两次的颜色才有贡献,所以记录一个ne[i]表示i后第一个和i同色的花,维护一个l,每次处理询问的时候l单调右移,树状数组维护,在ne[ne[i]]位置++,在ne[i]位置--(因为这里已经++过了),这样区间查询的时候如果右端点超过ne[ne[i]]就能对答案有贡献
注意在开始的时候给有ne的第一次出现的颜色的位置++

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=2000006;
int n,c,m,a
,ne
,ans
,t
,p
;
struct qwe
{
int l,r,id;
}q
;
bool cmp(const qwe &a,const qwe &b)
{
return a.l<b.l;
}
int read()
{
int r=0,f=1;
char p=getchar();
while(p>'9'||p<'0')
{
if(p=='-')
f=-1;
p=getchar();
}
while(p>='0'&&p<='9')
{
r=r*10+p-48;
p=getchar();
}
return r*f;
}
void update(int x,int v)
{
for(int i=x;i<=n;i+=(i&(-i)))
t[i]+=v;
}
int ques(int x)
{
int r=0;
for(int i=x;i>=1;i-=(i&(-i)))
r+=t[i];
return r;
}
int main()
{
n=read(),c=read(),m=read();
for(int i=1;i<=n;i++)
a[i]=read();
for(int i=n;i>=1;i--)
ne[i]=p[a[i]],p[a[i]]=i;
for(int i=1;i<=m;i++)
q[i].l=read(),q[i].r=read(),q[i].id=i;
sort(q+1,q+1+m,cmp);
for(int i=1;i<=c;i++)
if(p[i]&&ne[p[i]])
update(ne[p[i]],1);
for(int i=1,l=1;i<=m;i++)
{
for(;l<q[i].l;l++)
{
if(ne[l])
update(ne[l],-1);
if(ne[ne[l]])
update(ne[ne[l]],1);
}
ans[q[i].id]=ques(q[i].r)-ques(q[i].l-1);
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: