1483:[HNOI2009]梦幻布丁
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 2 通过数: 2
题目描述
N N 个布丁摆成一行,进行次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.
输入
第一行给出 N N ,表示布丁的个数和好友的操作次数. 第二行 N N 个数, A2 A 2 … An A n 表示第 i i 个布丁的颜色从第三行起有行,对于每个操作,若第一个数字是1表示要对颜色进行改变,其后的两个整数 X X ,表示将所有颜色为 X X 的变为, X X 可能等于. 若第一个数字为2表示要进行询问当前有多少段颜色,这时你应该输出一个整数. 0
输出
针对第二类操作即询问,依次输出当前有多少段颜色.
输入样例
4 3
1 2 2 1
2
1 2 1
2
输出样例
3
1
解:
乍一看是非常不可过的,但是确实是一道纯暴力题目。
给每种颜色开一个链表,合并的时候枚举长度小的,计算一下答案减少了多少,然后把两个合并。同时开一个数组记录每种颜色对应的编号即可。
这种小的并大的,合并期望复杂度是的。rand一下即可。
code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
int fa[1000005],head[1000005],nt[100005],col[100005];
int ans,n,m,x,y,fg;
void add(int op,int id){
nt[id]=head[op];
head[op]=id;
}
int main()
{
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&col[i]);
if(col[i]!=col[i-1]) ans++;
add(col[i],i);
fa[col[i]]=col[i];
}
for(int i=1;i<=m;i++){
scanf("%d",&fg);
if(fg==2){
printf("%d\n",ans);
continue;
}
scanf("%d%d",&x,&y);
if(fa[x]==0||x==y) continue;
if(fa[y]==0){
fa[y]=fa[x];fa[x]=0;
continue;
}
if((rand()&1)==0){
int j;
for(j=head[fa[x]];j!=-1;j=nt[j]){
if(col[j-1]==fa[y]) ans--;
if(col[j+1]==fa[y]) ans--;
}
for(int k=head[fa[x]];k!=-1;k=nt[k]) col[k]=fa[y],j=k;
nt[j]=head[fa[y]];head[fa[y]]=head[fa[x]];head[fa[x]]=-1;
fa[x]=0;
}
else{
int j;
for(j=head[fa[y]];j!=-1;j=nt[j]){
if(col[j-1]==fa[x]) ans--;
if(col[j+1]==fa[x]) ans--;
}
for(int k=head[fa[y]];k!=-1;k=nt[k]) col[k]=fa[x],j=k;
nt[j]=head[fa[x]];head[fa[x]]=head[fa[y]];head[fa[y]]=-1;
fa[y]=fa[x];fa[x]=0;
}
}
}