【题意】
N个布丁摆成一行,进行M次操作。每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色。
例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.。
n,m<=1000000
【思路】
对于每次操作,我们把小的合并到大的里面。
举个例子:颜色为1,2,2时要把颜色2改为颜色1,暴力的方法就是直接将所有颜色2改为1,但是将1改为2耗时更短。因此我们以f[i]表示i真正的颜色,那么若被替换的颜色数量大于将其替换的颜色数量,为了节省时间将替换颜色改为被替换颜色以后,要交换f[替换颜色]与f[被替换颜色]。
然后对于每种颜色都使用链表储存,将颜色1改为颜色2就将颜色1的结尾接到颜色2的头。
【代码】
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 1100000
using namespace std;
int n,m,ans,val[MAXN],f[MAXN],head[MAXN],st[MAXN],cnt[MAXN],nex[MAXN];
void solve(int x,int y)
{
cnt[y]+=cnt[x];
cnt[x]=0;
for(int i=head[x];i;i=nex[i])
{
if(val[i-1]==y)ans--;
if(val[i+1]==y)ans--;
}
for(int i=head[x];i;i=nex[i])
val[i]=y;
nex[st[x]]=head[y];
head[y]=head[x];
head[x]=0;
return ;
}
int main()
{
scanf("%d%d",&n,&m);
memset(head,0,sizeof(head));
memset(st,0,sizeof(st));
ans=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&val[i]);
f[val[i]]=val[i];
if(val[i]!=val[i-1])ans++;
if(!head[val[i]])st[val[i]]=i;
cnt[val[i]]++;
nex[i]=head[val[i]];
head[val[i]]=i;
}
while(m--)
{
int op,x,y;
scanf("%d",&op);
if(op==1)
{
scanf("%d%d",&x,&y);
if(x==y)continue;
if(cnt[f[x]]>cnt[f[y]])
swap(f[x],f[y]);
x=f[x],y=f[y];
if(!cnt[x])continue;
solve(x,y);
}
else printf("%d\n",ans);
}
return 0;
}