首页 > 代码库 > BZOJ 2120 数颜色(带修改莫队)
BZOJ 2120 数颜色(带修改莫队)
【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2120
【题目大意】
给出一颜色序列,每次可以修改一个位置的颜色或者询问一个区间不同颜色的数目。
【题解】
我们按照最近一次修改对查询操作进行时间标号,在莫队排序的时候引入时间作为第三维,
对于每个修改操作,我们记录修改之前的颜色和修改之后的颜色,
对于两个相邻两个查询,我们将其之间时间差距的修改操作进行记录或者撤销之后查询即可。
【代码】
#include <cstdio>#include <algorithm>#include <cmath>const int N=1000010;using namespace std;int Ans,ans[N],pos[N],c[N],pre[N],sz,cnt,n,m,limit;struct Q{ int l,r,id,t; friend bool operator < (const Q &a,const Q &b){ if(pos[a.l]!=pos[b.l])return pos[a.l]<pos[b.l]; if(a.r!=b.r)return a.r<b.r; return a.t<b.t; }}ask[N];struct update{int pos,c,pre;}w[N];int in[N],num[N];void cal(int x){ if(in[x]){if(!--num[c[x]])Ans--;} else{if(++num[c[x]]==1)Ans++;} in[x]^=1; } void update(int x,int y){ if(in[x]){cal(x);c[x]=y;cal(x);} else c[x]=y;}int main(){ while(~scanf("%d%d",&n,&m)){ int limit=sqrt(n); for(int i=1;i<=n;i++)scanf("%d",&c[i]),pre[i]=c[i],pos[i]=(i-1)/limit+1,in[i]=0; cnt=sz=Ans=0; for(int i=1;i<=m;i++){ char op[10]; scanf("%s",op); int x,y; scanf("%d%d",&x,&y); if(op[0]==‘R‘){w[++cnt].pos=x;w[cnt].c=y;w[cnt].pre=pre[x];pre[x]=y;} else{ask[++sz].l=x;ask[sz].r=y;ask[sz].id=sz;ask[sz].t=cnt;} }sort(ask+1,ask+sz+1); int l,r; cal(l=r=1); for(int i=1;i<=sz;i++){ for(int j=ask[i-1].t+1;j<=ask[i].t;j++)update(w[j].pos,w[j].c); for(int j=ask[i-1].t;j>ask[i].t;j--)update(w[j].pos,w[j].pre); while(l<ask[i].l)cal(l++); while(l>ask[i].l)cal(--l); while(r<ask[i].r)cal(++r); while(r>ask[i].r)cal(r--); ans[ask[i].id]=Ans; }for(int i=1;i<=sz;i++)printf("%d\n",ans[i]); }return 0;}
BZOJ 2120 数颜色(带修改莫队)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。