bzoj1483 [HNOI2009]梦幻布丁

1483: [HNOI2009]梦幻布丁

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 3485  Solved: 1376
[Submit][Status][Discuss]

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

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

 

 

首先把每一个颜色连成一条链，每次修改就直接把这条链上的颜色全部暴力修改，再看看前后位置的颜色是否与修改后的颜色相同，若相同就把答案减一。每次修改的时候只修改长度较短的那条链，再开个指针把这个颜色指向它实际在链表中的颜色即可。

#include<bits/stdc++.h>
using namespace std;
#define MAXN 1000000+10
int n,m,ans=0,col[MAXN],s[MAXN],st[MAXN],rl[MAXN],next[MAXN],head[MAXN];
void deal(int x,int y){
    for(int i=head[x];i;i=next[i]){
        if(col[i+1]==y)ans--;
        if(col[i-1]==y)ans--;
    }
    for(int i=head[x];i;i=next[i])col[i]=y;
    next[st[x]]=head[y];head[y]=head[x];
    head[x]=st[x]=0;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&col[i]);
        rl[col[i]]=col[i];
        if(col[i]!=col[i-1])ans++;
        if(!head[col[i]])st[col[i]]=i;
        s[col[i]]++;next[i]=head[col[i]];head[col[i]]=i;
    }
    while(m--){
        int opt,x,y;
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d%d",&x,&y);
            if(x==y)continue;
            if(s[rl[x]]>s[rl[y]])swap(rl[x],rl[y]);
            x=rl[x];y=rl[y];
            if(s[x]==0)continue;
            s[y]+=s[x];s[x]=0;
            deal(x,y);
        }
        else printf("%d\n",ans);
    }
    return 0;
}