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

数据范围没给……n、m<=10w

 

首先考虑用链表暴力合并每次复杂度是O(n)，总复杂度是O(nm)

然后用启发式合并的话，每次都把长度小的往长度大的那个套，每次复杂度是O(小的那个的长度)

但是我们这样想：每次我们用启发式合并，生成的序列的长度一定>=小的那个的长度的两倍

所以最多合并logn次

总复杂度nlogn

但是因为是启发式合并，它往哪一个合并是不知道的，所以还要用一个数组f[]保存第i个数合并之后变成什么数

唉八中oj真不科学为什么数组不够大是TLE啊……害得我找错找半天

#include<cstdio>
#define N 100010
#define M 1000010
int c[N],sum[M];
int next[N],last[M];
int head[M];
int f[M];
int n,m,ans;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
inline void move(int from,int to)
{
	for (int i=head[from];i;i=next[i])
	{
		if (c[i+1]==to)ans--;
		if (c[i-1]==to)ans--;
	}
	for (int i=head[from];i;i=next[i])c[i]=to;
	next[last[from]]=head[to];head[to]=head[from];
	sum[to]+=sum[from];sum[from]=head[from]=last[from]=0;
}
int main()
{
	n=read();m=read();
	for(int i=1;i<=n;i++)
	{
	  c[i]=read();f[c[i]]=c[i];
	  if (c[i]!=c[i-1])ans++;
	  if (!head[c[i]])last[c[i]]=i;
	  next[i]=head[c[i]];head[c[i]]=i;sum[c[i]]++;
	}
	int opr,x,y,a,b;
	for (int i=1;i<=m;i++)
	  {
	  	opr=read();
	  	if (opr==2){printf("%d\n",ans);continue;}
	  	x=read();y=read();
	  	if (x==y)continue;
	  	if (sum[f[x]]>sum[f[y]])
	  	{
	  		int t=f[x];
	  		f[x]=f[y];
	  		f[y]=t;
	  	}
	  	a=f[x];b=f[y];
	  	if (sum[a]==0)continue;
	  	move(a,b);
	  }
}