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

bzoj1483 [HNOI2009]梦幻布丁

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1483

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

数据范围
N，M<=1e6

题解：
启发式合并->时间复杂度O(mlogn)

首先用链表存储不同的颜色，这里使用了类似链式前向星的存储方式。记录每个颜色的链的起始位置，以及元素的个数。（便于之后合并）

为了实现启发式合并，我们必然把小的合并到大的中去。但是显然如果只是把颜色 i，颜色 j（size[i] < size[j]）都变成颜色j，而原操作是颜色 j 变成颜色 i，显然是不行的。因此，我们交换i，j所代表的颜色，从此i代表j，j代表 i，用一个数组维护。

这样思路就很清晰了，我们每次通过链表枚举数目较小的那个颜色的每个元素，统计看其旁边（左+右）有没有与要改成的颜色相接，若有，则ans- -。然后再一次通过链表枚举数目较小的那个颜色的每个元素，在原数组中改变他们的颜色。

关于合并，因为之前记录了每个颜色的链的起始位置，于是这样处理：

head[y] (new)-> head[x] -> … … … -> fir[x] ->head[y] (previous)

记得更新大颜色的size，并把被合并的颜色size和起始元素都清空。

关于启发式合并的复杂度分析：
显然合并的集合两个大小差距越悬殊，只枚举小的的元素的优势越明显。我们举最坏情况为例：
若每次合并的都是大小相同的两个集合，每一次的大小成两倍扩展，一个集合最多被扩展logn次，n*logn -> O(nlogn)

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1000006;
int n,m,ans=0,head[N],nxt[N],a[N],col[N],size[N],fir[N];
void merge(int x,int y) 
{
    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[fir[x]]=head[y];   
    head[y]=head[x];                                
    size[y]+=size[x];
    head[x]=size[x]=fir[x]=0;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]!=a[i-1]) ans++;
        col[a[i]]=a[i];
        if(!head[a[i]]) fir[a[i]]=i; 
        size[a[i]]++;
        nxt[i]=head[a[i]];
        head[a[i]]=i;
    }
    while(m--)
    {
        int opt,x,y;
        scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d",&x,&y);

            if(x==y) continue;
            if(size[col[x]]>size[col[y]]) swap(col[x],col[y]);
            x=col[x];y=col[y];
            if(size[x]==0) continue;
            merge(x,y);
        }
        else printf("%d\n",ans);
    }

    return 0;
}