MOOC浙大数据结构 — 06-图1 列出连通集 (25分)

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

输入格式:

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

输出格式:

按照"{ v_1v​1​​ v_2v​2​​ ... v_kv​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 <cstdio>
#include <vector>
#include <queue>
#include <algorithm>

#define MAX 10

using namespace std;

typedef struct {
    int vId;    // 相连接的结点编号
    int edge;   // 边权
} Node;

vector<int> MGraph[MAX];

int visit[MAX];

void dfs( int curNode ) {
    printf( " %d", curNode );
    for( int i = 0; i < MGraph[curNode].size(); i++ ) {
        int nextId = MGraph[curNode][i];
        if( !visit[nextId] ) {
            visit[nextId] = 1;
            dfs( nextId );
        }
    }
}

void bfs( int curNode, int n ) {
    queue<int> q;
    q.push( curNode );

    while( !q.empty() ) {
        int curId = q.front();
        q.pop();
        printf( " %d", curId );
        for( int i = 0; i < MGraph[curId].size(); i++ ) {
            int nextId = MGraph[curId][i];
            if( !visit[nextId] ) {
                q.push( nextId );
                visit[nextId] = 1;
            }
        }
    }
}


int main() {
    int N, M;
    scanf( "%d%d", &N, &M );

    int a, b;
    for( int i = 0; i < M; i++ ) {
        scanf( "%d%d", &a, &b );
        MGraph[a].push_back( b );
        MGraph[b].push_back( a );
    }

    for( int i = 0; i < N; i++ ) {
        if( MGraph[i].size() > 1 )
        sort( MGraph[i].begin(), MGraph[i].end() );
    }

    for( int i = 0; i < N; i++ ) {
        if( !visit[i] ) {
            visit[i] = 1;
            printf( "{" );
            dfs( i );
            printf( " }\n" );
        }
    }

    fill( visit, visit + N, 0 );

    for( int i = 0; i < N; i++ ) {
        if( !visit[i] ) {
            visit[i] = 1;
            printf( "{" );
            bfs( i, N );
            printf( " }\n" );
        }
    }
    return 0;
}