BZOJ 1483: [HNOI2009]梦幻布丁 启发式合并链表

1483: [HNOI2009]梦幻布丁

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 3424 Solved: 1350
[Submit][Status][Discuss]

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

HINT

Source

题解：
第一次写启发式合并链表。其实听【启发式】感觉很厉害，实际我理解的就是把长度小的往长度大的上合并。每次合并后短的长度至少会扩大到原来的短的一倍，而短的长度最多扩大logn次，就会出现在最长的链表中，（不会再动它），证明时间复杂度最多是O（nlogn）的。
【刚看到一篇博客，说可以理解为对每个节点最多操作logn次，所以均摊下来时间复杂度是O（nlogn）次。
对于这道题，我们可以先用一条链把所有颜色一样的串起来，然后要修改的时候，对于短一点的那条链，就直接暴力将它接在长链的后面。
但要多开一个数组记录每个数字代表的真实颜色，如果要将长的链上的颜色x并在短链y的上面，就将它们交换一下，这样下次查询的时候查到x替换成y，查到y替换成x，才不会弄错。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

const int N = 1000000 + 10;

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;
}

int n,m;
int ans=0,c[N],f[N],s[N];
int head[N],st[N],pre[N];

void work(int a,int b){
    for(int i=head[a];i;i=pre[i]){
        if(c[i+1]==b) ans--;
        if(c[i-1]==b) ans--;
    }
    for(int i=head[a];i;i=pre[i]) c[i]=b;
    pre[st[a]]=head[b];head[b]=head[a];
    head[a]=s[a]=st[a]=0;s[b]+=s[a];
}

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;++s[c[i]];
        if(!head[c[i]]) st[c[i]]=i;
        pre[i]=head[c[i]];head[c[i]]=i;
    }
    while(m--){
        int opt=read();
        if(opt==2) printf("%d\n",ans);
        else{
            int a=read(),b=read();
            if(a==b) continue;
            if(s[f[a]]>s[f[b]]) swap(f[a],f[b]);
            a=f[a],b=f[b];
            if(s[a]==0) continue;
            work(a,b);
        }
    }
    return 0;
}

set水过：
【其实我不太会用set，所以就当练习set了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;

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;
}

const int N = 1000000 + 10;

int n,m,ans=0;
int aa[N],f[N];
set<int> c[N];

void work(int a,int b){
    for(set<int>::iterator i=c[a].begin();i!=c[a].end();i++){
        if(aa[*i-1]==b) --ans;
        if(aa[*i+1]==b) --ans;
        c[b].insert(*i);
    }
    for(set<int>::iterator i=c[a].begin();i!=c[a].end();i++) aa[*i]=b;
    c[a].clear();
}

int main(){
    n=read(),m=read();
    for(register int i=1;i<=n;++i){
        aa[i]=read();c[aa[i]].insert(i);f[aa[i]]=aa[i];
        if(aa[i]!=aa[i-1]) ++ans;
    }
    while(m--){
        int opt=read();
        if(opt==2) printf("%d\n",ans);
        else{
            int a=read(),b=read();
            if(a==b) continue;
            if(c[f[a]].size()>c[f[b]].size()) swap(f[a],f[b]);
            a=f[a],b=f[b];
            work(a,b);
        }
    }
}