BZOJ 1483 HNOI 2009 梦幻布丁

本题是一道链表的题，由于n的值可能很大，所以就要用启发式合并来降低复杂度。

首先看到题目有两个操作，目的就是修改颜色，把一种颜色变为另一种，和询问颜色不同的段有多少个。我们发现统计颜色不同的段数其实只要看那个点的颜色是否和它的左右两边一样就行啦。所以一开始我们不妨预处理一下，用链表记下每种颜色在哪些位置和有多少个，再把一开始输入的种情况有多少种不同的段统计出来（用ans记录）。然后我们就要面对这题的难点，也就是修改问题啦。我们发现是不是和左右两边的一段其实就是看和两边颜色是否一样。如果交换想x和y两种颜色的，我们就先找出x和y中较小的那个链表。（这里用到啦启发式合并的思想，因为我们之后要合并两个链表所以得到的总链表长度一定大于最短的两倍，而我们每次操作的都是较小的所以最后整题的复杂度就可以从n^2降到nlogn级别）然后跑一遍，看那些点和它的左边右边颜色是否一样，不一样的话，这是我们先假设它和它颜色一样，则一旦不一样，说明前面加多啦就要把之前算的ans--就行啦。然后跑一遍把所有x的颜色变为y。最后再跑一遍如果此时点的颜色和左右两边不一样说明之前我们假设他们颜色是一样的是错误的，我们多减啦，所以此时要ans++，最后记住把x的链表挂到y的链表下面就ok啦。每次询问就直接输出ans就行啦。

代码如下：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <vector>
#include <queue>

#define M 1001011

using namespace std;

int n,m;
int a[M],num[M];
int next[M],p[M];
int father[M];
int ans=0;

void init()
{
    scanf("%d%d",&n,&m);
    memset(p,-1,sizeof(p));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        num[a[i]]++;
    }
    for(int i=1;i<=n;i++)
    {
        next[i]=p[a[i]];
        p[a[i]]=i;
    }
    for(int i=1;i<=n;i++)
        if(a[i]!=a[i-1])
            ans++;
    for(int i=1;i<=M-1;i++)
        father[i]=i;
}

void start()
{
    for(int j=1;j<=m;j++)
    {
        int ch,x,y;
        scanf("%d",&ch);
        if(ch==2)
            printf("%d\n",ans);
        else
        {
            scanf("%d%d",&x,&y);
            if(x==y)
                continue;
            if(num[father[x]]>num[father[y]])
                swap(father[x],father[y]);
            x=father[x];
            y=father[y];
            if(num[x]==0)
                continue;
            for(int i=p[x];i!=-1;i=next[i])
            {
                if(a[i]!=a[i-1])
                    ans--;
                if(a[i]!=a[i+1]&&i!=n)
                    ans--;
            }
            for(int i=p[x];i!=-1;i=next[i])
                a[i]=y;
            int head;
            for(int i=p[x];i!=-1;i=next[i])
            {
                if(a[i]!=a[i-1])
                    ans++;
                if(a[i]!=a[i+1]&&i!=n)
                    ans++;
                head=i;
            }
            num[y]+=num[x];
            num[x]=0;
            next[head]=p[y];
            p[y]=p[x];
            p[x]=-1;
        }
    }
}

int main()
{
    init();
    start();

    return 0;
}