描述:
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,2,2,1,三段颜色分别为 [1, 1], [2, 3], [4, 4]。一次操作后,布丁的颜色变为 1,1,1,1,只有[1,4] 一段颜色
输入样例
4 3
1 2 2 1
2
1 2 1
2
输出样例
3
1
Code:(看看就行了,为了完整性这代码是网上找的,绝对不会考)(C++)
(代码作者: 安月冷)
/*
vector 记录每个颜色出现的位置
修改时启发式合并,更细答案即可。
*/
#include<bits/stdc++.h>#define N 1000007
#define inf 0x3f3f3f3f
#define ll long longusing namespace std;
vector<int>v[N];
vector<int>::iterator it;
int n,m;
int fa[N],a[N];int main()
{scanf("%d%d",&n,&m);int ans=0;for(int i=1;i<=n;i++){scanf("%d",&a[i]);if(a[i]!=a[i-1])ans++;fa[a[i]]=a[i];v[a[i]].push_back(i);}while(m--){int t;scanf("%d",&t);if(t==2) printf("%d",ans);else{int x,z;scanf("%d%d",&x,&z);if(x==z)continue;if(v[fa[x]].size()>v[fa[z]].size()) swap(fa[x],fa[z]);int col1=fa[x],col2=fa[z];for(int i=0;i<v[col1].size();i++){v[col2].push_back(v[col1][i]);if(a[v[col1][i]-1]==col2) ans--;if(a[v[col1][i]+1]==col2) ans--;}for(int i=0;i!=v[col1].size();i++) a[v[col1][i]]=col2;v[col1].clear();}}return 0;
}