题目大意
其实这道题原题直译着实难理解,但是可以用一些形象的方式来理解:
给一个 n 个点的 Graph,第 i 个点一刚开始是第 I 种颜色,接着有 k 次操作,第 i 次操作有个参数 oi 代表颜色 oi 会侵略所有和自己相邻的颜色,于是所有和 oi 相邻的颜色全都变成 oi (若已没有颜色oi 已被侵略,则该次操作无效),求最终每个点的颜色。(自官方题解PPT)
解题思路
对于颜色,在这里可以用并查集维护,根结点表示颜色。
操作一种颜色时,与其相连的所有点颜色都会改变,所以对于每个点至多会有一次侵略操作。
使用一组vector,记录第i种颜色所相邻的点,每次操作时把x颜色相邻的每个点(将y分别赋值为a[x]中的点)所相邻的点集(a[y]中的点),转为x相邻的点集,y所在集合的颜色变为x。
AC代码
#include<bits/stdc++.h> using namespace std; vector<int> a[800010],b; int f[800010]; int find(int x) { if(x!=f[x]) f[x]=find(f[x]); return f[x]; } int main() { int t,n,m,q,x,y,z,i,j; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); for(i=0;i<=n;i++) f[i]=i,a[i].clear(); for(i=1;i<=m;i++) { scanf("%d%d",&x,&y); a[x].push_back(y); a[y].push_back(x); } scanf("%d",&q); while(q--) { scanf("%d",&x); if(f[x]!=x) continue; b=a[x];a[x].clear(); for(i=0;i<b.size();i++) { y=find(b[i]); if(y!=x) { f[y]=x; if(a[x].size()<a[y].size()) swap(a[x],a[y]); for(j=0;j<a[y].size();j++) a[x].push_back(a[y][j]); } } } for(i=0;i<n;i++) printf("%d ",find(i)); printf("\n"); } return 0; }