Bzoj1483 [HNOI2009]梦幻布丁

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 2655  Solved: 1063

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

HINT

1<=n,m<=100,000; 0<Ai,x,y<1,000,000

Source

 

看到数据范围比较怂，然而看到题解都是用挺暴力的方法刚过去的。想了想好像有点道理。

 

链表+启发式合并

 

用链表存储每种颜色的所有布丁的位置，每次修改时，合并链表和颜色标记，记录链表长度，暴力修改长度较小的那个链表。

然而写挂了。向题解低头之后发现没有记录布丁原来的颜色，导致染色出了偏差。

 

↓注意37和38行的语句差别。要根据原色判断是否该染。

#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
using namespace std;
const int mxn=2000100;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m;
int hd[mxn],nxt[mxn],cnt[mxn];
int now[mxn];
int co[mxn];
int main()
{
    int i,j;
    n=read();m=read();
    int ans=0;
    for(i=1;i<=n;i++){
        co[i]=read();
        cnt[co[i]]++;
        nxt[i]=hd[co[i]];
        now[co[i]]=co[i];
        hd[co[i]]=i;
        if(co[i]!=co[i-1])ans++;
    }
    int x,y;
    for(i=1;i<=m;i++){
        x=read();
        if(x==2){printf("%d\n",ans);}
        else{
            x=read();y=read();
            if(x==y)continue;
//          if(cnt[x]>cnt[y])swap(x,y);
            if(cnt[now[x]]>cnt[now[y]])swap(now[x],now[y]);
            x=now[x];y=now[y];//原色 
            if(!cnt[x])continue;
            int last=0;
            for(j=hd[x];j;j=nxt[j]){
                if(co[j-1]==y)ans--;
                if(co[j+1]==y)ans--;
                last=j;
            }
            for(j=hd[x];j;j=nxt[j]){
                co[j]=y;
            }
            nxt[last]=hd[y];
            hd[y]=hd[x];
            hd[x]=0;
            cnt[y]+=cnt[x];
            cnt[x]=0;
        }
    }
    return 0;
}