1483：[HNOI2009]梦幻布丁（链表+启发式合并）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/lvmaooi/article/details/82694415

1483：[HNOI2009]梦幻布丁

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 2 通过数: 2

题目描述

$N$个布丁摆成一行,进行$M$次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.

输入

第一行给出$N$,$M$表示布丁的个数和好友的操作次数. 第二行$N$个数$A_1$,$A_2$…$A_n$表示第$i$个布丁的颜色从第三行起有$M$行,对于每个操作,若第一个数字是1表示要对颜色进行改变，其后的两个整数$X$,$Y$表示将所有颜色为$X$的变为$Y$，$X$可能等于$Y$. 若第一个数字为2表示要进行询问当前有多少段颜色，这时你应该输出一个整数. 0

输出

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

输入样例

4 3
1 2 2 1
2
1 2 1
2

输出样例

3
1

解：

乍一看是非常不可过的，但是确实是一道纯暴力题目。
给每种颜色开一个链表，合并的时候枚举长度小的，计算一下答案减少了多少，然后把两个合并。同时开一个数组记录每种颜色对应的编号即可。
这种小的并大的，合并期望复杂度是$nlog$的。rand一下即可。

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
int fa[1000005],head[1000005],nt[100005],col[100005];
int ans,n,m,x,y,fg;

void add(int op,int id){
    nt[id]=head[op];
    head[op]=id;
}

int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&col[i]);
        if(col[i]!=col[i-1]) ans++;
        add(col[i],i);
        fa[col[i]]=col[i];
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&fg);
        if(fg==2){
            printf("%d\n",ans);
            continue;
        }
        scanf("%d%d",&x,&y);
        if(fa[x]==0||x==y) continue;
        if(fa[y]==0){
            fa[y]=fa[x];fa[x]=0;
            continue;
        }
        if((rand()&1)==0){
            int j;
            for(j=head[fa[x]];j!=-1;j=nt[j]){
                if(col[j-1]==fa[y]) ans--;
                if(col[j+1]==fa[y]) ans--;
            }
            for(int k=head[fa[x]];k!=-1;k=nt[k]) col[k]=fa[y],j=k;
            nt[j]=head[fa[y]];head[fa[y]]=head[fa[x]];head[fa[x]]=-1;
            fa[x]=0;
        }
        else{
            int j;
            for(j=head[fa[y]];j!=-1;j=nt[j]){
                if(col[j-1]==fa[x]) ans--;
                if(col[j+1]==fa[x]) ans--;
            }
            for(int k=head[fa[y]];k!=-1;k=nt[k]) col[k]=fa[x],j=k;
            nt[j]=head[fa[x]];head[fa[x]]=head[fa[y]];head[fa[y]]=-1;
            fa[y]=fa[x];fa[x]=0;
        }
    }
}