p3201&bzoj1483 梦幻布丁
题目
Input
Output
针对第二类操作即询问,依次输出当前有多少段颜色.
Sample Input
1 2 2 1
2
1 2 1
2
Sample Output
3
1
分析
用链表将每种颜色的部分连在一起,每次修改利用启发式合并的思想将个数少的颜色合到个数多的颜色,因为这样可能改变颜色对应关系,所有我们用一个数组tr表示这个数所对应的原来的颜色是什么,每一次根据换颜色后是否跟左右颜色一样来更新答案。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<vector>
using namespace std;
int n,m,cnt;
int col[1000010],c[1000010],size[1000010];
int ans;
int be[1000010],head[1000010],tail[1000010],tr[1000010];
map<int,int>mp;
int main(){
int i,j,k;
cin>>n>>m;
for(i=1;i<=n;i++){
cin>>col[i];
tr[col[i]]=col[i];
if(col[i]!=col[i-1]){
c[++cnt]=col[i],size[col[i]]++;
if(!tail[col[i]])tail[col[i]]=cnt;
be[cnt]=head[col[i]];
head[col[i]]=cnt;
}
}
ans=cnt;
for(i=1;i<=m;i++){
cin>>k;
if(k==2)cout<<ans<<endl;
else {
int x,y;
cin>>x>>y;
if(x==y)continue;
if(size[tr[y]]<size[tr[x]])swap(tr[x],tr[y]);
x=tr[x];
y=tr[y];
if(!size[x])continue;
for(j=head[x];j;j=be[j]){
if(c[j-1]==y){
ans--;
}
if(c[j+1]==y){
ans--;
}
}
for(j=head[x];j;j=be[j])
c[j]=y;
size[y]+=size[x];
size[x]=0;
be[tail[x]]=head[y];
head[y]=head[x];
head[x]=tail[x]=0;
}
}
return 0;
}