BZOJ1483 [HNOI2009]梦幻布丁（洛谷P3201）

启发式合并

BZOJ题目传送门
洛谷题目传送门

对每个颜色记录其出现的所有位置，每次改变颜色的时候把小的并到大的上面并更新答案。这样可能大的是要改变的颜色，那么再记录一下当前修改的实际上是哪个颜色就好了。

记录位置可以用链表。我用了vector。

注意清数组

代码：

#include<vector>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define M 1000005
#define F inline
using namespace std;
int n,m,mx,ans,p[M],c[M]; vector <int> a[M];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    return l==r?EOF:*l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void mrg(int x,int y){
    if (a[p[x]].size()>a[p[y]].size()) swap(p[x],p[y]); x=p[x],y=p[y];
    if (x==y) return; int nx=a[x].size(),ny=a[y].size();
    for (int i=0;i<nx;i++){
        int tmp=a[x][i]; ans-=(c[tmp-1]==y)+(c[tmp+1]==y);
    }
    for (int i=0;i<nx;i++) c[a[x][i]]=y,a[y].push_back(a[x][i]);
    a[x].clear();
}
int main(){
    n=_read(),m=_read();
    for (int i=1;i<=n;i++)
        a[c[i]=_read()].push_back(i),ans+=c[i]!=c[i-1],mx=max(mx,c[i]);
    for (int i=1;i<M;i++) p[i]=i;
    for (int x,y;m;m--)
        if (_read()==2) printf("%d\n",ans);
        else x=_read(),y=_read(),mrg(x,y);
    return 0;
}