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
想了一个nm的线段树无果考虑乱搞
每次染色其实可以直接交换一下,让总数小的染成大的那个,然后记录一下实际颜色即可
在乱搞的过程中用链表进行优化,硬拆合并,效率感人
原来这就叫启发式合并
代码如下:
//MT_LI
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
int n,m,ans;
int v[1110000],a[1110000];
int tot[1110000];
int next[1110000],pre[1110000];
int main()
{
    scanf("%d%d",&n,&m);
    ans=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]!=a[i-1])ans++;
        next[i]=pre[a[i]];
        pre[a[i]]=i;
        tot[a[i]]++;v[a[i]]=a[i];
    }
    while(m--)
    {
        int op;
        scanf("%d",&op);
        if(op==1)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(x==y)continue;
            if(tot[v[x]]>tot[v[y]])swap(v[x],v[y]);
            if(!pre[v[x]])continue;
            int pos=0;
            for(int i=pre[v[x]];i;i=next[i])
            {
                pos=i;
                if(a[i+1]==v[y])ans--;
                if(a[i-1]==v[y])ans--;
            }
            for(int i=pre[v[x]];i;i=next[i])a[i]=v[y];
            next[pos]=pre[v[y]];pre[v[y]]=pre[v[x]];
            pre[v[x]]=0;
            tot[v[y]]+=tot[v[x]];tot[v[x]]=0;
        }
        else printf("%d\n",ans);
    }
    return 0;
}

 

posted @ 2018-10-24 10:05  MT_LI  阅读(151)  评论(0编辑  收藏  举报