bzoj1483: [HNOI2009]梦幻布丁【链表+启发式合并】

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。
n,m<=1000000

Output

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

Sample Input

4 3

1 2 2 1

2

1 2 1

2

Sample Output

3

1

解题思路：

注意到一种颜色变为另一种颜色后就只能一起操作了，所以可以用链表+启发式合并，合并时维护全局答案即可。
但是血WA，其实要注意hzwer所提到的：
比如把1染成2，但是s[1]>s[2]，这时我们应该将2合并到1的链后面，但是会遇到一个麻烦的问题，就是这个链头是接1下的，也就是说以后找颜色2，发现没有颜色2只有颜色1。。。
于是我们应该开一个数组f，表示我们寻找一种颜色时，实际应该找哪个颜色下的链，遇到上面那种情况要交换f[1]和f[2]。

#include<bits/stdc++.h>
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=1000005;
int n,m,ans,a[N],id[N],st[N],first[N],nxt[N],len[N];

void merge(int x,int y)
{
    for(int i=first[x];i;i=nxt[i])
    {
        if(a[i+1]==y)ans--;
        if(a[i-1]==y)ans--;
    }
    for(int i=first[x];i;i=nxt[i])a[i]=y;
    nxt[st[x]]=first[y],first[y]=first[x];
    len[y]+=len[x],st[x]=len[x]=first[x]=0;
}

int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),m=getint();
    for(int i=1;i<=n;i++)
    {
        a[i]=getint(),id[a[i]]=a[i];
        if(!st[a[i]])st[a[i]]=i;
        if(a[i]!=a[i-1])ans++;
        nxt[i]=first[a[i]],first[a[i]]=i,len[a[i]]++;
    }
    while(m--)
    {
        int op=getint();
        if(op==2)printf("%d\n",ans);
        else
        {
            int x=getint(),y=getint();
            if(x==y)continue;
            if(len[id[y]]<len[id[x]])swap(id[x],id[y]);
            x=id[x],y=id[y];
            if(len[x]==0)continue;
            merge(x,y);
        }
    }
    return 0;
}