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


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


Sample Input


4 3


1 2 2 1


2


1 2 1


2


Sample Output


3


1

将所有一样的颜色用链表连接，先计算出初始答案，当每次修改时相当于将两个不同的颜色合并。但是一定要将个数少的颜色合并到个数多的颜色的后面，这样可以保证每次新链的长度一定>=较小的子链的两倍。相当于每次长度*2这样最多也就修改logn次，因此复杂度是nlogn。

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define swap(x,y){int temp=x;x=y;y=temp;}
const int maxm = 1000005;
int last[maxm], ne[maxm], s[maxm], id[maxm], a[maxm], p[maxm], fa[maxm];
int main()
{
	int n, i, j, k, sum, m, ord, ans = 0;
	scanf("%d%d", &n, &m);
	for (i = 1;i <= n;i++)
	{
		scanf("%d", &a[i]);
		if (a[i] != a[i - 1]) ans++;
		if (!fa[a[i]]) fa[a[i]] = i;
		id[a[i]] = i;s[a[i]]++;
		ne[i] = last[a[i]];
		last[a[i]] = i;p[a[i]] = a[i];
	}
	while (m--)
	{
		scanf("%d", &ord);
		if (ord == 1)
		{
			int x, y;
			scanf("%d%d", &x, &y);
			if (x == y) continue;
			if (s[p[x]] > s[p[y]]) swap(p[x], p[y]);
			x = p[x], y = p[y];
			if (!s[x]) continue;
			for (i = last[x];i >= 1;i = ne[i])
			{
				if (a[i - 1] == y) ans--;
				if (a[i + 1] == y) ans--;
			}
			for (i = last[x];i >= 1;i = ne[i])
				a[i] = y;
			s[y] += s[x];s[x] = 0;
			ne[fa[x]] = last[y];last[y] = last[x];
			last[x] = fa[x] = 0;
		}
		else
			printf("%d\n", ans);
	}
	return 0;
}