bzoj 1483: [HNOI2009]梦幻布丁

1483: [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

Output

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

Sample Input

4 3
1 2 2 1
2
1 2 1
2

Sample Output

3
1

ACTY真大！！！

———————以下题解———————

链表加优化。。

每种颜色一条链

我们使用启发式合并，每次把元素少的链合并到元素多的链上去，但是有一个问题就是题目规定输入x，y，把x颜色的变为y。

假如颜色x的元素多于y，我们还是可以把y合并到x，只是再维护一个father数组，表示颜色x实际上表示什么。

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int M=1000005;
const int N=100005;
int n,m,i,ans,x,y,p,a[N],s[M],head[M],last[M],Next[N],f[M];
int main()
{
    scanf("%d%d",&n,&m);
    ans=0;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        f[a[i]]=a[i];
        s[a[i]]++;
        Next[i]=last[a[i]];
        last[a[i]]=i;
        if(!head[a[i]]) head[a[i]]=i;
        if(a[i]!=a[i-1]) ans++;
    }
    while(m--)
    {
        scanf("%d",&p);
        if(p==1)
        {
            scanf("%d%d",&x,&y);
            if(x==y) continue;
            if(s[f[x]]>s[f[y]]) swap(f[x],f[y]);
            x=f[x];y=f[y];
            if(!s[x]) continue;
            for(i=last[x];i;i=Next[i])
            {
                if(a[i-1]==y) ans--;
                if(a[i+1]==y) ans--;
            }
            for(i=last[x];i;i=Next[i]) a[i]=y;
            s[y]+=s[x];
            s[x]=0;
            Next[head[y]]=last[x];
            head[y]=head[x];
        } else printf("%d\n",ans);
    }
    return 0;
}