PTA 5-6 列出连通集 (25)

题目：http://pta.patest.cn/pta/test/15/exam/4/question/714

PTA - 数据结构与算法题目集（中文）- 5-6

给定一个有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 <iostream>
#include <queue>
#define MAXSIZE 10
using namespace std;
//注：(*a)[4]：a为指针，指向有四个元素的数组

void DFS(int (*data)[MAXSIZE], bool visited[], int i)
{
    if(visited[i]==true)
        return;
    cout << i << " ";
    visited[i]=true;

    for(int j=0;j<MAXSIZE;j++)
    {
        if(data[i][j]==1 && visited[j]==false)
            DFS(data,visited,j);
    }
}

void BFS(int (*data)[MAXSIZE], bool visited[], int i)
{
    if(visited[i]==true)
        return;
    queue<int> myQueue;
    myQueue.push(i);
    visited[i]=true;

    while(myQueue.size()!=0)
    {
        //!取队头元素
        int x=myQueue.front();
        cout << x << " ";
        myQueue.pop();
        for(int j=0;j<MAXSIZE;j++)
        {
            //!相连且未被访问过
            if(data[x][j]==1 && visited[j]==false)
            {
                myQueue.push(j);
                visited[j]=true;
            }
        }
    }
}

int main()
{
    int data[MAXSIZE][MAXSIZE]={0};    //!用邻接矩阵存储图
    bool visited_DFS[MAXSIZE]={false};
    bool visited_BFS[MAXSIZE]={false};

    int vertices,edges;
    cin >> vertices >> edges;

    for(int i=0;i<edges;i++)
    {
        int a,b;
        cin >> a >> b;
        data[a][b]=1;  //!使a,b相连
        data[b][a]=1;
    }

    //!深度优先遍历，从数字小的结点开始遍历
    for(int i=0; i<vertices; i++)
    {
        if(visited_DFS[i]==false)
        {
            cout << "{ ";
            DFS(data, visited_DFS, i);
            cout << "}
";
        }
    }

    //!广度优先遍历，从数字小的结点开始遍历
    for(int i=0; i<vertices; i++)
    {
        if(visited_BFS[i]==false)
        {
            cout << "{ ";
            BFS(data, visited_BFS, i);
            cout << "}
";
        }
    }
    return 0;
}