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
1 2 2 1
2
1 2 1
2
Sample Output
3
1
1
题解:
用链表维护一下每种颜色的位置,合并的时候使用启发式合并把个数小的接在个数大的后面。
代码:
#include<iostream>
#include<cstdio>
#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);
}
}
}