bzoj 1438(启发式合并)

1483: [HNOI2009]梦幻布丁

Time Limit: 10 Sec   Memory Limit: 64 MB
Submit: 2341   Solved: 965
[ 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

Output

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

Sample Input

4 3
1 2 2 1
2
1 2 1
2

Sample Output

3

1

解题思路：完全没有想到可以启发式合并。以后遇到

合并的题可以考虑。这道题就是用启发式合并链，

由于是少的插到多的，所以还要记录每个颜色查询时

真正所对的颜色。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,ans;
int c[300000],fa[1000100],next[300000],head[1000100],sz[1000100];


inline int read()
{
char y; int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
} 


void merge(int u,int v)
 {
  for (int i=head[u];i;i=next[i]) ans-=(c[i-1]==v)+(c[i+1]==v);
  for (int i=head[u];i;i=next[i]) c[i]=v;
  for (int i=head[u];i;i=next[i]) if (!next[i]) {next[i]=head[v];break;}
head[v]=head[u]; head[u]=0; sz[v]+=sz[u]; sz[u]=0;
 }
 
int main()
{
n=read(); m=read(); ans=0;
for (int i=1;i<=n;++i)
{
c[i]=read(); fa[c[i]]=c[i]; ++sz[c[i]];
if (c[i]!=c[i-1])++ans;
        next[i]=head[c[i]]; head[c[i]]=i;
}
for (int i=1;i<=m;++i)
{
int g=read();
if (g==1)
{
  int x,y; x=read(); y=read();
  if (x==y) continue;
  if (sz[fa[x]]>sz[fa[y]]) swap(fa[x],fa[y]);
  x=fa[x]; y=fa[y];   
  if (!sz[x]) continue;
  merge(x,y);
}else  
 {
  printf("%d\n",ans);
 }
}
}