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
1 2 2 1
2
1 2 1
2
Sample Output
3
1
本地AC，BZOJ上RE，唉。
水题，注意不要把set定义在结构体里，大数据会错，有个陷阱就是x==y。
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <set>
using namespace std;
const int N=;
set<int>::iterator it1,it2;
int c[N],id[N*],s[N];
int n,Q,ans,tot;set<int>v[N];
void Insert(int p,int i){
it2=v[p].lower_bound(i);
if(it2!=v[p].end()&&i+==*it2)s[p]-=;
if(it2!=v[p].begin()&&i-==*(--it2))s[p]-=;
v[p].insert(i);s[p]+=;
} int main(){
freopen("pudding.in","r",stdin);
freopen("pudding.out","w",stdout);
ios::sync_with_stdio(false);
cin.tie(NULL),cout.tie(NULL);
cin>>n>>Q;
for(int i=;i<=n;i++)cin>>c[i];
for(int i=;i<=n;i++){
int &p=id[c[i]];
if(!p)p=++tot;
Insert(p,i);
} for(int i=;i<=tot;i++)ans+=s[i];
int tp,x,y,px,py;
while(Q--){
cin>>tp;
if(tp==){
cin>>x>>y;if(x==y)continue;
if(v[id[x]].size()>v[id[y]].size())swap(id[x],id[y]);
int px=id[x],py=id[y];ans-=s[px]+s[py];
for(it1=v[px].begin();it1!=v[px].end();it1++)
Insert(py,*it1);
ans+=s[py];s[px]=;v[px].clear();
}
if(tp==)cout<<ans<<"\n";
}
return ;
}