题目:
题解:
我们可以给每一种颜色建立链表,因为链上的颜色总是一起变的,那么修改的时候只需要将链表合并就行了,把小的合并到大的里面去,那么均摊的复杂度就是O(nlogn)
动态的维护答案,当合并的时候看看左看看右,如果是要变成一样的颜色就答案- -
这里还有一个小细节,我们把小的合并进入大的,但并不一定是小的的颜色变成大的颜色。怎么办呢?我们设置一个id[i]表示i这个颜色目前对应的颜色,然后如果我们要把大的的颜色改成小的,就把两个id换一下就好了
还要注意因为链表是前向星要特别注意head和tail。每次变颜色的时候要先看看需要改变的颜色是不是没有,如果没有就不需要改
代码:
#include <cstdio>
#include <iostream>
using namespace std;
const int N=1000005;
int a[N],tail[N],f[N],cnt[N],head[N],nxt[N];
int main()
{
int n,m,ans=0,maxx=0;scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
{
scanf("%d",&a[i]);maxx=max(maxx,a[i]);
if (a[i]!=a[i-1]) ans++;
if (!cnt[a[i]]) tail[a[i]]=i;
nxt[i]=head[a[i]]; head[a[i]]=i;
cnt[a[i]]++;
}
for (int i=1;i<=maxx;i++) f[i]=i;
while (m--)
{
int id,x,y;
scanf("%d",&id);
if (id==1)
{
scanf("%d%d",&x,&y);
if (f[x]==f[y]) continue;
if (cnt[f[x]]>cnt[f[y]]) swap(f[x],f[y]);
x=f[x]; y=f[y];
if (cnt[x]==0) continue;///
for (int i=head[x];i;i=nxt[i])
{
if (a[i-1]==y) ans--;
if (a[i+1]==y) ans--;
}
for (int i=head[x];i;i=nxt[i]) a[i]=y;
nxt[tail[x]]=head[y]; head[y]=head[x]; head[x]=tail[x]=0;
cnt[y]+=cnt[x]; cnt[x]=0;
}
else printf("%d\n",ans);
}
}