[HNOI2009]梦幻布丁 D1 T1_第三行起有m行,对于每个操作,若第一个数字是1表示要对颜色进行改变,其后的两个整-优快云博客

本文介绍了一道关于布丁颜色变换的问题，通过一系列操作改变布丁颜色并计算不同颜色段的数量。采用数据结构和算法优化解决方案，适用于算法竞赛训练。

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 31 2 2 121 2 12

Sample Output

HINT

0 < n, m < 100001

0 < Ai, x, y < 106

Source

HNOI2009

#include<stdio.h>
int n,m,a[100001],f[1000001],s[1000001],k,c,d,ans,head[1000001],next[1000001],l[1000001];
void merge(int x,int y)
{
	for(int i=head[x];i;i=next[i])
	{
		if(a[i+1]==y)
			ans--;
		if(a[i-1]==y)
			ans--;	
	}
	for(int i=head[x];i;i=next[i])
		a[i]=y;
	next[l[x]]=head[y];
	head[y]=head[x];
	s[y]+=s[x];
	s[x]=0;
	head[x]=0;
	l[x]=0;	
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		f[a[i]]=a[i];
		if(a[i]!=a[i-1])
			ans++;
		if(head[a[i]]==0)
			l[a[i]]=i;
		s[a[i]]++;
		next[i]=head[a[i]];
		head[a[i]]=i;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&k);
		if(k==1)
		{
			scanf("%d%d",&c,&d);
			if(c==d)
				continue;
			if(s[f[c]]>s[f[d]])
			{
				int z=f[d];
				f[d]=f[c];
				f[c]=z;
			}
			if(s[f[c]]==0)
				continue;
			s[f[d]]+=s[f[c]];
			s[f[c]]=0;
			merge(f[c],f[d]);
		}
		else
		{
			printf("%d\n",ans);
		}
	}	
}