【BZOJ1483】[HNOI2009]梦幻布丁
Description
N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.
Input
第一行给出N,M表示布丁的个数和好友的操作次数. 第二行N个数A1,A2...An表示第i个布丁的颜色从第三行起有M行,对于每个操作,若第一个数字是1表示要对颜色进行改变，其后的两个整数X,Y表示将所有颜色为X的变为Y，X可能等于Y. 若第一个数字为2表示要进行询问当前有多少段颜色，这时你应该输出一个整数. 0
Output
针对第二类操作即询问，依次输出当前有多少段颜色.
Sample Input
4 3
1 2 2 1
2
1 2 1
2
1 2 2 1
2
1 2 1
2
Sample Output
3
1
1
题解：我们对于每个颜色，用一个链表（或vector，或queue，或~）来维护它所有出现的位置，然后合并两个颜色时就相当于将两个链表合并，顺便记录一下有多少相邻的位置，更新一下答案即可。合并时用启发式合并即可做到nlogn。
数据范围：好像n<=1e5，颜色<=1e6，m<=1e6
特判：没有颜色为Y的情况，没有颜色为X的情况，X=Y的情况。
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=100010;
const int maxm=1000010;
int n,m,tot,cnt,ans;
int c[maxn],head[maxn],next[maxn],to[maxn];
int bel[maxm],siz[maxn],p[maxn];
int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<'0'||gc>'9') {if(gc=='-')f=-f; gc=getchar();}
while(gc>='0'&&gc<='9') ret=ret*10+gc-'0',gc=getchar();
return ret*f;
}
void add(int a,int b)
{
next[b]=head[a],head[a]=b;
}
int main()
{
n=rd(),m=rd();
int i,j,a,b,aa,bb;
for(i=1;i<=n;i++)
{
a=rd();
if(!bel[a]) bel[a]=++tot;
c[i]=bel[a],siz[bel[a]]++,add(bel[a],i);
if(c[i]!=c[i-1]) ans++;
}
for(i=1;i<=m;i++)
{
if(rd()==2) printf("%d
",ans);
else
{
a=rd(),b=rd();
if(a==b) continue;
if(!bel[a]) continue;
if(!bel[b])
{
bel[b]=bel[a],bel[a]=0;
continue;
}
aa=a,bb=b,a=bel[a],b=bel[b],bel[aa]=0;
if(siz[a]>siz[b]) swap(a,b);
bel[bb]=b;
for(p[0]=0,j=head[a];j;j=next[j]) p[++p[0]]=j,ans-=(c[j-1]==b)+(c[j+1]==b);
for(j=1;j<=p[0];j++) add(b,p[j]),c[p[j]]=b;
head[a]=0;
}
}
return 0;
}