启发式合并训练之一

以前没怎没接触过，快要考试了，赶快来补一补

https://www.luogu.org/problem/P3201

分析：

将同一种颜色用类似链式前向星的形式存起来

考虑启发式合并，将小的合并到大的上面

如果要将x变为y，

如果sz[x]<sz[y],就直接把x接在y的上面就好

如果sz[x]>sz[y],如果直接把y接在x后面颜色就会变为x

所以我们需要使用一个 f数组，表示当我们要寻找颜色 x 时，实际上需要寻找颜色为 f[x] 的链。如果遇到上面这种情况就要交换交换 f[x]和 f[y]。

code ：

#include<bits/stdc++.h>
#define ll long long
#define il inline
#define ri register int
#define lowbit(x) x&(-x)
using namespace std;
const int maxn=1e6+5;
int head[maxn],nxt[maxn],fir[maxn];
int a[maxn],col[maxn],sz[maxn];
int n,m,ans;
il void merge(int x,int y){
    for(ri i=head[x];i;i=nxt[i]){
        if(a[i-1]==y)ans--;
        if(a[i+1]==y)ans--;
    }
    for(ri i=head[x];i;i=nxt[i])a[i]=y;
    nxt[fir[x]]=head[y];
    head[y]=head[x]; 
    sz[y]+=sz[x];
    head[x]=sz[x]=fir[x]=0;
}
int main(){
    scanf("%d%d",&n,&m);
    for(ri i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(a[i]!=a[i-1])ans++;
       col[a[i]]=a[i];
       if(!head[a[i]])fir[a[i]]=i;
       sz[a[i]]++;
       nxt[i]=head[a[i]];
       head[a[i]]=i;
    }
    while(m--){
        int op,x,y;
        scanf("%d",&op);
       if(op==1){
        scanf("%d%d",&x,&y);
        if(x==y)continue;
        if(sz[col[x]]>sz[col[y]])swap(col[x],col[y]);
        if(!sz[col[x]])continue;
        merge(col[x],col[y]);
    }
    else printf("%d\n",ans);
    }
    return 0;
}