题目描述
题解
首先搞出来一颗生成树。如果存在多个连通块,把它们都连到1上去。
对于每一条非树边,连上它形成一个环,那么这个环上的所有点一定是在同一个联盟里的。
于是我们可以用并查集将树链上的点缩起来,在做并查集的过程中维护点的大小。由于每个点都只会被合并一次,时间复杂度是
O(n)
的。
代码
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 200005
int n,m,p;
int tot,point[N],nxt[N*4],v[N*4];
struct hp{int x,y,pd;}edge[N*2];
int h[N],father[N],f[N],size[N];
int find(int x)
{
if (x==f[x]) return x;
f[x]=find(f[x]);
return f[x];
}
void addedge(int x,int y)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
father[x]=fa; h[x]=h[fa]+1;
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa)
dfs(v[i],x);
}
void merge(int x,int y)
{
x=find(x),y=find(y);
while (x!=y)
{
if (h[x]<h[y]) swap(x,y);
int f1=find(x),f2=find(father[f1]);
if (f1!=f2)
{
f[f1]=f2;
size[f2]+=size[f1];
}
x=f2;
}
}
int main()
{
scanf("%d%d%d",&n,&m,&p);
for (int i=1;i<=m;++i) scanf("%d%d",&edge[i].x,&edge[i].y);
for (int i=1;i<=p;++i) scanf("%d%d",&edge[m+i].x,&edge[m+i].y);
for (int i=1;i<=n;++i) f[i]=i;
for (int i=1;i<=m+p;++i)
{
int f1=find(edge[i].x),f2=find(edge[i].y);
if (f1!=f2)
{
edge[i].pd=0;
addedge(edge[i].x,edge[i].y);
f[f1]=f2;
}
else edge[i].pd=1;
}
for (int i=1;i<=n;++i)
{
int f1=find(1),f2=find(i);
if (f1!=f2)
{
f[f2]=f1;
addedge(1,f2);
}
}
dfs(1,0);
for (int i=1;i<=n;++i) f[i]=i,size[i]=1;
for (int i=1;i<=m;++i)
if (edge[i].pd)
{
int f1=find(edge[i].x),f2=find(edge[i].y);
if (f1!=f2) merge(f1,f2);
}
for (int i=m+1;i<=m+p;++i)
if (edge[i].pd)
{
int f1=find(edge[i].x),f2=find(edge[i].y);
if (f1!=f2) merge(f1,f2);
printf("%d\n",size[find(f1)]);
}
else puts("No");
}
总结
①利用并查集缩点的思路之前没有见过,要注意一下。包括也要注意一下时间复杂度分析。