1483: [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

针对第二类操作即询问，依次输出当前有多少段颜色.

题解：

看到网上题解都是什么启发式合并，蒟蒻并不会QAQ。

只能用set水过。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<set>
using namespace std;
const int N=1000005;
int n,m;
int a[N];
int tr[N],ans=0;
set<int>q[N];
void slove(int a1,int b)
{
	for(set<int>::iterator i=q[a1].begin();i!=q[a1].end();i++)
	{
		if(a[*i-1]==b) ans--;
		if(a[*i+1]==b) ans--;
		q[b].insert(*i);
	} 
	for(set<int>::iterator i=q[a1].begin();i!=q[a1].end();i++)a[*i]=b;
	q[a1].clear();
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		tr[a[i]]=a[i];
		if(a[i]!=a[i-1]) ans++;
		q[a[i]].insert(i);
	}
	int yu,x,x1,y1;
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&x);
		if(x==2){printf("%d\n",ans);}
		else
		{
			scanf("%d%d",&x1,&y1);
			if(x1==y1) continue;
			if(q[tr[x1]].size()>q[tr[y1]].size())//谁换谁对答案并不会有影响 
			swap(tr[x1],tr[y1]);
			int tx=tr[x1],ty=tr[y1];
			slove(tx,ty);
		}
	}
}

昨晚看到社长和理事长虐了一晚上狗QAQ，今早去饭堂又看到他俩。

所以说我为什么要进摄影社QAQ