PTA_数据结构与算法_06-图1 列出连通集

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

输入格式:

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

输出格式:

按照"{ v1 v2 ... 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 <iostream>
#include <queue>
#include <cstring>

using namespace std;

const int N = 11;
int n, m;
bool visited[N];
int g[N][N];
queue<int> q;

void dfs(int u)
{
    visited[u] = true;
    cout << " " << u;
    for (int i = 0; i < n; i ++)
        if (g[u][i] == 1 && !visited[i])
            dfs(i);
}

void bfs(int u)
{
    visited[u] = true;
    cout << " " << u;
    q.push(u);
    
    while(q.size())
    {
        int t = q.front();
        q.pop();
        for (int i = 0; i < n; i ++)
            if (g[t][i] == 1 && !visited[i])
            {
                visited[i] = true;
                cout << " " << i;
                q.push(i);
            }
    }
}


int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i ++)
    {
        int x, y;
        cin >> x >> y;
        g[x][y] = 1;
        g[y][x] = 1;
    }

    for (int i = 0; i < n; i ++)
    {
        if (!visited[i])
        {
            cout << "{";
            dfs(i);
            cout << " }" << endl;
        }
    }

    memset(visited, false, sizeof visited);

    for (int i = 0; i < n; i ++)
    {
        if (!visited[i])
        {
            cout << "{";
            bfs(i);
            cout << " }" << endl;
        }
    }
}