找朋友c++

找朋友

题目描述:

有朋友路好走，每个人都需要很多朋友。朋友的朋友也是朋友，如和𝐴和 𝐵是朋友，和𝐵和 𝐶是朋友，那么和𝐴和 𝐶当然是朋友。

现在给出各对朋友的关系，请把小明（1号）的朋友全部都找出来，请注意朋友的朋友也是朋友。

输入格式:

输出一个整数𝑀,𝑁，表示总共有𝑀个同学，这些同学中有𝑁对是朋友的关系。

接下来有N行，每两行两个整数𝑥,𝑦，表示 和𝑥和 𝑦 是朋友的关系。

输出格式:

一个整数，表示1号同学的朋友总数。

样例输入:

10 6
1 2
3 4
1 10
2 3
5 6
6 9

样例输出:

4

提示:

6 对给定的朋友关系中，我们能找到和1号同学是朋友的人数总共是 4 人，分别是 2 、3、4、10。

#include <bits/stdc++.h>
using namespace std;
int m,n,x,y,a[5100][5100],ans,b[5100];

void dfs( int dep ) {

    for(int i=1;i<=n;i++){
        if(a[dep][i]==1){
            if(b[i]==0){
            b[i]=1;
            ans++;
        }
            a[dep][i]=0;
            
            dfs(i);
        }
    }
    for(int i=1;i<=n;i++){
        if(a[i][dep]==1){
            if(b[i]==0){
            b[i]=1;
            ans++;
        }
            a[dep][i]=0;
        
            dfs(i);
        
        }
    }
    return;

}
int main() {
    cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        cin >> x >> y;
        a[x][y] = a[y][x] =1;
    }
    dfs( 1 );
    cout << ans-1;
    return 0;
}