拼題A 列出連通集 (25分)

題目描述

給定一個有N個頂點和E條邊的無向圖，請用DFS和BFS分別列出其所有的連通集。假設頂點從0到N−1編號。進行搜索時，假設我們總是從編號最小的頂點出發，按編號遞增的順序訪問鄰接點。

輸入格式:

輸入第1行給出2個整數N(0<N≤10)和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 <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>

using namespace std;
const int MAX = 11;
int N, E;
int a[MAX][MAX], vis[MAX];

void BFS(int u)
{
    queue<int> QUE;
    while( !QUE.empty() )
        QUE.pop();

    QUE.push(u);
    vis[u] = 1;

    while( !QUE.empty() )
    {
        int t = QUE.front();
        QUE.pop();
        printf("%d ", t);
        for( int i = 0; i < N; i++ )
        {
            if(a[t][i] && !vis[i])
            {
                QUE.push(i);
                vis[i] = 1;
            }
        }
    }

}

void DFS(int u)
{
    if(u >= N)
        return;
    printf("%d ", u);
    vis[u] = 1;
    for(int i = 0; i < N; i++)
    {
        if(a[u][i] == 1 && !vis[i])
        {
            DFS(i);
        }
    }
}

int main()
{
    scanf("%d %d", &N, &E);
    int x, y;
    memset(a, 0, sizeof(a));

    for(int i = 0; i < E; i++)
    {
        scanf("%d %d", &x, &y);
        a[x][y] = 1;
        a[y][x] = 1;
    }

    memset(vis, 0, sizeof(vis));
    for(int i = 0; i < N; i++)
    {
        if(!vis[i])
        {
            printf("{ ");
            DFS(i);
            printf("}\n");
        }
    }

    memset(vis, 0, sizeof(vis));
    for(int i = 0; i < N; i++)
    {
        if(!vis[i])
        {
            printf("{ ");
            BFS(i);
            printf("}\n");
        }
    }
    return 0;
}