和小哥哥一起刷洛谷(5)

图论之深度优先搜索DFS

Posted by ethan-zhou on September 20, 2018

关于dfs

dfs伪代码:

void dfs(s){
    for(int i=0;i<s的出度;i++){
        if(used[i]为真) continue;
        used[i]=1;
        dfs(i);
    }
    return;
}

统计无向图的连通分量

显然,你在洛谷上是搜不到这题的,因为这是我们学校团队的题。所以还是找个小板凳专心听我讲吧。

题目描述:

给定无向图G(V,E),请统计G中连通分量的数量。

  • 连通分量:结点V的一个子集V’,保证V’中任意两点间都有路径
  • 需要在主循环中进行多次dfs

输入输出格式:

输入格式:

第一行包含两个整数N、M,表示该图共有N个结点和M条无向边(N<= 5000,M<=200000);

接下来M行,每行包含2个整数{u,v},表示有一条无向边(u,v)。

输出格式:

一个整数,代表图G连通分量的数量

样例:

输入:

5 4
1 5
2 3
3 4
4 2

输出:

2

代码:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<iostream>
#include<queue>
#include<vector>
using namespace std;
const int NR=5005;
bool color[NR];//used数组
int cnt=0,n,m;
vector<int> link[NR];
void dfs(int a){//dfs函数
    int sz=link[a].size();
    for(int i=0;i<sz;i++){
        int nx=link[a][i];
        if(color[nx]==false){
            color[nx]=true;
            dfs(nx); 
        }
    }
    return;
} 
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        int st,en;
        scanf("%d%d",&st,&en);
        link[st].push_back(en);
        link[en].push_back(st);
    }
    for(int i=1;i<=n;i++){//对于每个没有去过的点,将其所有可以到达的点标为true,计数加一,重复
        if(color[i])continue;
        color[i]=true;
        dfs(i);
        cnt++;
    }
    cout<<cnt;
    return 0;
}

作者:@ethan-enhe
本文为作者原创,转载请注明出处:本文链接