【bzoj2453】维护队列 分块

来源:互联网 发布:凹陷性疤痕 知乎 编辑:程序博客网 时间:2024/05/01 13:58

强行分块,明明主席树简单得多。

莫队也不错。

每个数记录一个pre[i]表示上次出现的位置,那么询问[l,r]中有多少个不同的数等价于[l,r]中有多少个pre[i]<l

主席树做就好了,分块也很简单。


#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<algorithm>#include<iostream>#define maxn 10010using namespace std;int a[maxn],b[maxn];int l[110],r[110],bel[maxn];int c[maxn],last[1000100];char s[10];int n,m,block,tot,T;int find(int L,int R,int x){int l=L,r=R,ans=L-1;while (l<=r){int mid=(l+r)/2;if (b[mid]<x) ans=mid,l=mid+1;else r=mid-1;}return ans-L+1;}int query(int x,int y){int ans=0;if (bel[x]==bel[y]){for (int i=x;i<=y;i++) if (a[i]<x) ans++;}else{for (int i=bel[x]+1;i<=bel[y]-1;i++) ans+=find(l[i],r[i],x);for (int i=x;i<=r[bel[x]];i++) if (a[i]<x) ans++;for (int i=l[bel[y]];i<=y;i++) if (a[i]<x) ans++;}return ans;}void rebuild(int x){for (int i=l[x];i<=r[x];i++) b[i]=a[i];sort(b+l[x],b+r[x]+1);}void modify(int x,int d){for (int i=1;i<=n;i++) last[c[i]]=0;c[x]=d;for (int i=1;i<=n;i++){int t=a[i];a[i]=last[c[i]];if (a[i]!=t) rebuild(bel[i]);last[c[i]]=i; }}int main(){scanf("%d%d",&n,&m);block=sqrt(n);tot=(n-1)/block+1;for (int i=1;i<=n;i++){int Block=(i-1)/block+1;if (!l[Block]) l[Block]=i;r[Block]=i;bel[i]=Block;}for (int i=1;i<=n;i++) {scanf("%d",&c[i]);a[i]=last[c[i]];last[c[i]]=i;}for (int i=1;i<=n;i++) b[i]=a[i];for (int i=1;i<=tot;i++) sort(b+l[i],b+r[i]+1);for (int i=1;i<=m;i++){scanf("%s",s);if (s[0]=='Q'){int x,y;scanf("%d%d",&x,&y);printf("%d\n",query(x,y));}else{int x,c;scanf("%d%d",&x,&c);modify(x,c);}}return 0;}


0 0
原创粉丝点击