图论模板 连通分量数 DFS or 并查集

图论模板 连通分量数 DFS or 并查集

C Battle Over Cities (25 分)（DFS）

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=2000005;//注意：链式前向星，无向图，开两倍（最大边数1000*1000的两倍）！！！ 
int n,m,k,a,b,c,ans=0;//ans为连通分量数量 
struct edge
{
	int to,next;
}edge[num];
int cnt,head[num];
void init()
{
	for(int i=0;i<num;i++)
	{
		edge[i].next=-1;
		head[i]=-1;
	}
	cnt=0;
}
void addedge(int u,int v)
{
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}//链式前向星存图 

int vis[1010],flag;
void dfs(int x)
{
	for(int i=head[x];i!=-1;i=edge[i].next)
		if(edge[i].to!=c)
		{
			if(vis[edge[i].to]==0)
			{
				vis[edge[i].to]=1;
				dfs(edge[i].to);
			}
		}
}

int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m>>k;
	init();
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b;
		addedge(a,b);
		addedge(b,a);
	}
	for(int i=1;i<=k;i++)
	{
		memset(vis,0,sizeof(vis));
		ans=0;
		cin>>c;
		for(int i=1;i<=n;i++)
		{
			if(vis[i]==0&&i!=c)
			{
				vis[i]=1;
				ans++;
				dfs(i);
			}
		}
		cout<<ans-1<<endl;
	}
	return 0;
}

C Battle Over Cities (25 分)（并查集）

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=1010;
vector<int> G[maxn];//邻接表 
int n,m,k,a,b,c,p[maxn],vis[maxn];
int find(int x)
{
	return p[x]==x?x:p[x]=find(p[x]);
}//并查集 
void unio(int a,int b)
{
	int x=find(a),y=find(b);
	if(x!=y) p[x]=y;
}
void init()
{
	for(int i=1;i<=n;i++)
	{
		p[i]=i;
		vis[i]=0;
	}
}
int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m>>k;
	for(int i=0;i<m;i++)
	{
		cin>>a>>b;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	for(int t=0;t<k;t++)
	{
		cin>>c;
		init();
		for(int i=1;i<=n;i++)
		{
			int size=G[i].size();
			for(int j=0;j<size;j++)
			{
				if(i!=c&&G[i][j]!=c)
					unio(i,G[i][j]);
			}
		}
		int ans=0;
		/* 
		for(int i=1;i<=n;i++)
			if(i!=c) find(i);//注意是find(i)不是p[i] 
		sort(p+1,p+n+1);
		for(int i=1;i<=n;i++)
			if(p[i]!=p[i-1]&&p[i]!=c)
				ans++;
		cout<<ans-1<<endl;
		*/
		for(int i=1;i<=n;i++)
		{
			if(i==c) continue;
			int fa=find(i);
			if(!vis[fa])
			{
				ans++;
				vis[fa]=1;
			}
		}
		cout<<ans-1<<endl;
	}
	return 0;
}

F Interstellar Love（DFS）

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=20005;

struct edge
{
	int to,next;
}edge[num];
int cnt,head[num];
void init()
{
	for(int i=0;i<num;i++)
	{
		edge[i].next=-1;
		head[i]=-1;
	}
	cnt=0;
}
void addedge(int u,int v)
{
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}

int vis[10005],flag;
void dfs(int x,int fa)
{
	for(int i=head[x];i!=-1;i=edge[i].next)
		if(edge[i].to!=fa)
		{
			if(vis[edge[i].to]==0)
			{
				vis[edge[i].to]=1;
				dfs(edge[i].to,x);
			}
			else
				flag=0;
		}
}

int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int T;
	cin>>T;
	for(int t=1;t<=T;t++)
	{
		int n,m,a,b,ans=0,fix=0;
		cin>>n>>m;
		init();
		memset(vis,0,sizeof(vis));
		for(int i=1;i<=m;i++)
		{
			cin>>a>>b;
			addedge(a,b);
			addedge(b,a);
		}
		for(int i=1;i<=n;i++)
			if(vis[i]==0)
			{
				vis[i]=1;
				flag=1;
				if(head[i]!=-1)
				{
					ans++;
					dfs(i,0);
					if(flag==0)
						fix++;
				}
			}
		printf("Night sky #%d: %d constellations, of which %d need to be fixed.\n\n",t,ans,fix);
	}
	return 0;
}

F Interstellar Love（并查集）

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=20005;
int vis[num],fanode[num],faedge[num];//点的祖先节点，边的祖先节点 
int havenode[num],haveedge[num];//祖先节点连接点数，祖先节点连接边数 
int find(int x)
{
	return x==fanode[x]?x:fanode[x]=find(fanode[x]);
}
int main()
{
	int T;
	cin>>T;
	for(int t=1;t<=T;t++)
	{
		int n,m,a,b,ans=0,fix=0;
		cin>>n>>m;
		for(int i=1;i<=n;i++)
		{
			fanode[i]=i;
			vis[i]=havenode[i]=haveedge[i]=0;
		}
		for(int i=1;i<=m;i++)
		{
			cin>>a>>b;
			vis[a]=vis[b]=1;
			int x=find(a),y=find(b);
			if(x!=y)
				fanode[x]=fanode[y];
			faedge[i]=fanode[y];
		}
		for(int i=1;i<=n;i++)
			havenode[find(fanode[i])]++;
		for(int i=1;i<=m;i++)
			haveedge[find(faedge[i])]++;
		//for(int i=1;i<=n;i++)
		//	cout<<havenode[i]<<" "<<haveedge[i]<<endl;
		for(int i=1;i<=n;i++)
		{
			if(vis[i]&&havenode[i])
			{
				ans++;//至少有两个结点的连通分量 
				if(havenode[i]<=haveedge[i])
					fix++;//存在环的连通分量 
			}
		}
		printf("Night sky #%d: %d constellations, of which %d need to be fixed.\n\n",t,ans,fix);
	}
	return 0;
}