启发式合并,在合并时将siz较小的合并到siz较大的后面
题意:
有两个操作
1、将所有的x改成y
2、求序列的颜色段 如1 2 2 1 ans=2(m<=1000000)
首先,可以求出初始序列的ans,然后每次修改颜色相当于合并两个颜色,启发式合并。
启发式合并就是将sz小(s1)的暴力加入sz大的(s2),因为sz[s1]至少翻倍,至多翻倍至totsz,所以复杂度是O(logn)
附上并不知道正确性的代码
#include<cstdio> #include<cmath> #include<algorithm> #define maxn 1000010 using namespace std; int n,m,ans,a[maxn],head[maxn],nex[maxn],f[maxn],sz[maxn]; void insert(int col,int wher){ int tmp=head[col];head[col]=wher;nex[wher]=tmp; } void merge(int &x,int &y){ x=f[x];y=f[y]; if(x==y)return; if(sz[x]>sz[y])swap(x,y); if(head[x]==0)return; for(int i=head[x];i;i=nex[i]){ if(a[i+1]==y)ans--; if(a[i-1]==y)ans--; } for(int i=head[x];i;i=nex[i])a[i]=y; int k; for(k=head[y];k;k=nex[k]); nex[k]=head[x];sz[y]+=sz[x];sz[x]=head[x]=0; } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); if(a[i]!=a[i-1])ans++; f[a[i]]=a[i];sz[a[i]]++; insert(a[i],i); } while(m--){ int typ; scanf("%d",&typ); if(typ==1){int x,y;merge(x,y);}else printf("%d\n",ans); } }