列出连通集

110 阅读1分钟

在这里插入图片描述
普通的深搜和广搜问题。

#include<iostream>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
int a[100],n;
int N,E;
vector<vector<int>> edges;
void DFS(int v,bool *visited)
{
	a[n++]=v;
	visited[v]=true;
	for(int i=0;i<N;i++)
	{
		if(!visited[i]&&edges[v][i]==1)
		DFS(i,visited);
	}
}
void DFSTraverse()
{
	bool *visited=new bool[N];
	for(int i=0;i<N;i++)
	visited[i]=false;
	for(int i=0;i<N;i++)
	{
		memset(a,0,sizeof(a));
		n=0;
		if(!visited[i]) {
			DFS(i,visited);
			cout<<"{ ";
			for(int j=0;j<n;j++)
			{
				cout<<a[j];
				if(j!=n-1) cout<<" ";
			}
			cout<<" }"<<endl;
		}
	}
	delete []visited;
}
void BFSTraverse()
{
	queue<int> q;
	bool *visited=new bool[N];
	for(int i=0;i<N;i++)
	visited[i]=false;
	for(int i=0;i<N;i++)
	{
		memset(a,0,sizeof(a));
	    n=0;
		if(!visited[i])
		{
			visited[i]=true;
			q.push(i);
			a[n++]=i;
			while(!q.empty())
			{
				int u=q.front();
				q.pop();
				for(int j=0;j<N;j++)
				{
					if(edges[u][j]==1&&!visited[j])
					{
						a[n++]=j;
						visited[j]=true;
						q.push(j);
					}
				}
			}
			cout<<"{ ";
			for(int k=0;k<n;k++)
			{
				cout<<a[k];
				if(k!=n-1) cout<<" ";
			}
			cout<<" }"<<endl;
		}
	}
	delete []visited;
}
int main()
{
	cin>>N>>E;
	edges.resize(N);
	for(int i=0;i<N;i++)
	{
		edges[i].resize(N);
	}
	for(int i=0;i<N;i++)
	{
		for(int j=0;j<N;j++)
		{
			edges[i][j]=0;
		}
	}
	for(int i=0;i<E;i++)
	{
		int va,vb;
		cin>>va>>vb;
		edges[va][vb]=edges[vb][va]=1;
	}
	DFSTraverse();
	BFSTraverse();
	return 0;
}