【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

Sample Output

3
1

题解：

用链表维护一下每种颜色的位置，合并的时候使用启发式合并把个数小的接在个数大的后面。

代码：

#include
#include
#define M 1000010
int point[M],next[M],size[M],f[M],n,k,ans,c[M],h[M],x,y,m;
using namespace std;
void work(int x,int y)
{
	for (int i=point[x];i;i=next[i])
	 {
	 	if (c[i-1]==y) ans--;
		if (c[i+1]==y) ans--; 
	 }
	for (int i=point[x];i;i=next[i]) c[i]=y;
	next[h[x]]=point[y];point[y]=point[x];
	h[x]=point[x]=0;
}
int main()
{
  scanf("%d%d",&n,&m);
  for (int i=1;i<=n;i++)
   {
     scanf("%d",&c[i]);
     f[c[i]]=c[i];size[c[i]]++;
     if (c[i]!=c[i-1]) ans++;
     if (!point[c[i]]) h[c[i]]=i;
     next[i]=point[c[i]];point[c[i]]=i;
   }
  for (int i=1;i<=m;i++)
   {
	 scanf("%d",&k);
     if (k==2) printf("%d\n",ans);
     else
      {
        scanf("%d%d",&x,&y);
		if (x==y) continue;
		if (size[f[x]]>size[f[y]]) swap(f[x],f[y]);
		x=f[x];y=f[y];
		if (size[x]==0) continue;
		size[y]+=size[x];size[x]=0;
		work(x,y); 
      }
   }
}