[D-OJ練習] 求一個無向圖的連通分量(兩種寫法)

已知無向圖的頂點為字元型，要求采用鄰接矩陣表示，圖中頂點序号按字元順序排列，從鍵盤輸入圖中頂點的個數、邊的條數、頂點的資訊和邊的組成等。(注意：判斷一個無向圖是否連通) 求一個無向圖的連通分量。

輸入描述

第一行輸入無向圖的頂點數和邊的條數，以空格隔開

第二行輸入每個頂點的資料，中間沒有空格

第三行輸入每條邊，每條邊的格式為i j，中間有空格，所有邊占一行

輸出描述

輸出該無向圖的連通分量，占一行

輸入樣例

5 5

ABCDE

0 1 0 4 1 2 2 3 3 4

輸出樣例

DFS寫法

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e5+10;
int visit[N],h[N], e[N], ne[N], idx;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs(int u)
{
    visit[u] = true; //點u已經被周遊過

    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!visit[j]) dfs(j);
    }
}

int main()
{
    int n,m;
    cin >> n >> m;
    
    string s;
    cin >> s;
    
    memset(h, -1, sizeof h);
    int a, b;
    for(int i = 0; i < m; i++){
        cin >> a >> b;
        add(a, b), add(b, a);
    }
    
    int res = 0;
    for(int i = 0; i < n; i++){
        if(!visit[i]){
            res ++;
            dfs(i);     
        }
    }
    cout << res;
    return 0;
}      

BFS寫法

#include <iostream>
#include <cstring> 
using namespace std;

const int N = 1e5+10;
int h[N], e[N], ne[N], idx;
int visit[N];
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void bfs(int u)
{
    int q[N];
    int front = -1, rear = -1;
    q[++rear] = u;
    visit[u] = 1;
    
    while(front != rear)
    {
        int t = q[++front];
        for(int i = h[t]; i != -1; i = ne[i]){
            int j = e[i];
            if(!visit[j]){
                visit[j] = 1;
                q[++rear] = j;
            }
        }
    }
}
int main()
{
    int n,m;
    cin >> n >> m;
    
    string s;
    cin >> s;
    
    memset(h, -1, sizeof h);
    int a, b;
    for(int i = 0; i < m; i++){
        cin >> a >> b;
        add(a, b), add(b, a);
    }
    
    int res = 0;
    for(int i = 0; i < n; i++){   
        if(!visit[i]){
            res ++;
            bfs(i);     
        }
    }
    cout << res;
    return 0;
}