PTA | 06-图1 列出连通集 (25分)

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

输入格式:

输入第1行给出2个整数N(0<N≤10)和E，分别是图的顶点数和边数。随后E行，每行给出一条边的两个端点。每行中的数字之间用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<bits/stdc++.h>
using namespace std;
#define MAX 10
int a[MAX][MAX], N, E;
bool dfs_book[MAX], bfs_book[MAX];
queue<int>q;
void dfs(int c) {
    dfs_book[c] = true;
    printf(" %d", c);
    for (int i = 0 ;i< N; ++i) {
        //利用二维数组的一行就是该节点的邻接点，如果那个邻接点还没没访问过则递归访问
        if (a[c][i] && !dfs_book[i])
            dfs(i);
    }
}
void bfs(int c) {
    bfs_book[c] = 1;         //1表示访问过了
    q.push(c);
    printf(" %d", c);
    while (!q.empty()) {  //如果队列不空则每次从队列中取出一个节点找出该节点的第一层bfs节点，并加入队列中
        int temp = q.front();
        q.pop();
        for (int i = 0; i < N; i++) {     //找出第一层bfs的节点,依次输出并加入队列，跟树的层次遍历很像
            if (a[temp][i] && !bfs_book[i]) {
                printf(" %d", i);
                bfs_book[i] = 1;
                q.push(i);
            }
        }
    }
}

int main(){
    int temp1, temp2;
    scanf("%d%d", &N, &E);
    for (int i = 0; i < E; i++) {
        scanf("%d%d", &temp1, &temp2);
        a[temp1][temp2] = 1;        //因为是无向图所有邻接矩阵是关于主对角线对称的
        a[temp2][temp1] = 1;
    }
    for (int i = 0; i < N; i++) {
        if (!dfs_book[i]) {
            putchar('{');
            dfs(i);
            printf(" }
");
        }
    }

    for (int i = 0; i < N; i++) {
        if (!bfs_book[i]) {
            putchar('{');
            bfs(i);
            printf(" }
");
        }
    }
}