7-6 列出连通集 (25 分)

7-6 列出连通集 (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<bits/stdc++.h>
using namespace std;

vector<int> G[1005];
int vis[1005];

int DFS(int num){
    stack<int> s;
    s.push(num);
    vis[num] = 1;
    printf("{ %d ", num);
    while(s.size() > 0){
        bool flag = true;
        int x = s.top();
        for(int i = 0; i < G[x].size(); i++){
            if(vis[G[x][i]] == 0) {
                printf("%d ", G[x][i]);
                s.push(G[x][i]);
                vis[G[x][i]] = 1;
                flag = false;
                break;
            }
        }
        if(flag){
            s.pop();
        }
    }
    printf("}\n");
    return 0;
}

int BFS(int num){
    queue<int> q;
    q.push(num);
    vis[num] = 1;
    printf("{ %d ", num);
    while(!q.empty()){
        int x = q.front();
        q.pop();
        for(int i = 0; i < G[x].size(); i++){
            if(!vis[G[x][i]]){
                q.push(G[x][i]);
                printf("%d ", G[x][i]);
                vis[G[x][i]] = 1;
            }
        }
    }
    printf("}\n");
    return 0;
}

int main(){
    int a, b;
    scanf("%d %d", &a, &b);
    //读取边的连通情况
    while( b -- ){
        int x, y;
        scanf("%d %d", &x, &y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
    //把邻接表都从小到大排序
    for(int i = 0; i < a; i++ ) sort(G[i].begin(), G[i].end());
    memset(vis, 0, sizeof(vis));
    for (int i = 0; i < a; ++i) {
        if(!vis[i])
            DFS(i);
    }
    memset(vis, 0, sizeof(vis));
    for (int i = 0; i < a; ++i) {
        if(!vis[i])
            BFS(i);
    }
    return 0;
}