洛谷P3201 [HNOI2009]梦幻布丁

题目描述

N个布丁摆成一行,进行M次操作.每次将某个颜色的布丁全部变成另一种颜色的,然后再询问当前一共有多少段颜色.例如颜色分别为1,2,2,1的四个布丁一共有3段颜色.

输入输出格式

输入格式：

第一行给出N,M表示布丁的个数和好友的操作次数. 第二行N个数A1,A2...An表示第i个布丁的颜色从第三行起有M行,对于每个操作,若第一个数字是1表示要对颜色进行改变，其后的两个整数X,Y表示将所有颜色为X的变为Y，X可能等于Y. 若第一个数字为2表示要进行询问当前有多少段颜色，这时你应该输出一个整数. 0

输出格式：

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

输入输出样例

输入样例#1： 复制

4 3
1 2 2 1
2
1 2 1
2

输出样例#1： 复制

3
1

说明

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

 

用N个平衡树维护这N个颜色出现的位置

就本题而言，完全可以用一个set水过

每次合并的时候暴力合并就可以

注意当读入的颜色相同的时候直接跳出

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<ctime>
 #include<cstdlib>
 #include<algorithm>
 #include<set>
 using namespace std;
 #define ls T[now].ch[0]
 #define rs T[now].ch[1]
 const int MAXN=1e6+;
 inline char nc()
 {
     static char buf[MAXN],*p1=buf,*p2=buf;
     return p1==p2&&(p2=(p1=buf)+fread(buf,,MAXN,stdin),p1==p2)?EOF:*p1++;
 }
 inline int read()
 {
     char c=nc();int x=,f=;
     while(c<''||c>''){if(c=='-')f=-;c=nc();}
     while(c>=''&&c<=''){x=x*+c-'',c=nc();}
     return x*f;
 }
 set<int>s[MAXN];
 int a[MAXN],f[MAXN],ans;
 void unionn(int x,int y)
 {
     for(set<int>::iterator i=s[x].begin();i!=s[x].end();i++)
     {
         if(a[*i-]==y)  ans--;
         if(a[*i+]==y)    ans--;
         s[y].insert(*i);
     }
     for(set<int>::iterator i=s[x].begin();i!=s[x].end();i++)
         a[*i]=y;
     s[x].clear();
 }
 int main()
 {
     #ifdef WIN32
     freopen("a.in","r",stdin);
     #else
     #endif
     int n=read(),m=read();
     for(int i=;i<=n;i++)
     {
         a[i]=read();
         if(a[i]!=a[i-]) ans++;
         f[a[i]]=a[i];
         s[a[i]].insert(i);
     }
     while(m--)
     {
         int opt=read();
         if(opt==)    { printf("%d\n",ans);continue;}
         int a=read(),b=read();
         if(a==b)    continue;
         if(s[f[a]].size()>s[f[b]].size())    swap(f[a],f[b]);
         unionn(f[a],f[b]);
     }
     return ;
 }