题目大意
其实这道题原题直译着实难理解,但是可以用一些形象的方式来理解:
给一个 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;
}

京公网安备 11010502036488号