7-6 列出连通集

Posted 2021-10-24 cbattle

https://pintia.cn/problem-sets/15/problems/714

给定一个有N个顶点和E条边的无向图，请用DFS和BFS分别列出其所有的连通集。假设顶点从0到N−1编号。进行搜索时，假设我们总是从编号最小的顶点出发，按编号递增的顺序访问邻接点。

输入格式:

输入第1行给出2个整数N(0)和E，分别是图的顶点数和边数。随后E行，每行给出一条边的两个端点。每行中的数字之间用1空格分隔。

输出格式:

按照"{ v?1?? v?2?? ... v?k?? }"的格式，每行输出一个连通集。先输出DFS的结果，再输出BFS的结果。

输入样例:

8 6
0 7
0 1
2 0
4 1
2 4
3 5

输出样例:

{ 0 1 4 2 7 }
{ 3 5 }
{ 6 }
{ 0 1 2 7 4 }
{ 3 5 }
{ 6 }

提交代码

#include <stdio.h>
#include <iostream>
using namespace std;
#include <queue>

#define MAXN 11

bool arr[MAXN][MAXN] = {0};
bool dfsFlag[MAXN] = {0};
bool bfsFlag[MAXN] = {0};
bool dfsFirst = 1;
bool bfsFirst = 1;
int pointNum,edgeNum;

void DFS(int point)
{
    dfsFlag[point] = 1;
    printf(" %d",point);
    for(int i=0;i<pointNum;i++)
    {
        if(dfsFlag[i]==0 && arr[point][i]==1)
        {
            DFS(i);
        }
    }
}

void BFS()
{
    queue<int> q;
    for(int i=0;i<pointNum;i++)
    {
        if(bfsFlag[i]==0)
        {
            printf("{");
            bfsFlag[i] = 1;
            q.push(i);
            printf(" %d",i);
            while(q.empty()==false)
            {
                int front = q.front();
                q.pop();
                for(int j=0;j<pointNum;j++)
                {
                    if(arr[front][j]==1 && bfsFlag[j]==0)
                    {
                        bfsFlag[j] = 1;
                        printf(" %d",j);
                        q.push(j);
                    }
                }
            }
            printf(" }\n");
        }
    }
}

int main()
{
    scanf("%d %d",&pointNum,&edgeNum);
    int from,to;
    for(int i=0;i<edgeNum;i++)
    {
        scanf("%d %d",&from,&to);
        arr[from][to] = 1;
        arr[to][from] = 1;
    }
    
    //DFS
    for(int i=0;i<pointNum;i++)
    {
        if(dfsFlag[i]==0)
        {
            printf("{");        
            DFS(i);
            printf(" }\n");
        }
    }
    
    //BFS
    BFS();
    
    return 0;
}