[BFS] 5567번 결혼식

5567번 결혼식

 

https://www.acmicpc.net/problem/5567

 

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다.

 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

 

예제 입력

6
5
1 2
1 3
3 4
2 3
4 5

 

예제 출력

3

 

해결방법

단순 그래프 BFS 탐색 문제이다

 

시작 노드로부터의 거리가 1이거나 2인 노드의 개수를 구해야한다

ans = 0;
for(int i=1; i<=n; i++){
    if(dist[i]==1 || dist[i]==2) ans++;
}
cout << ans << "\n";

 

 

소스코드

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <bitset>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#define MAX 501
#define INF 987654321
using namespace std;

int n,m,u,v,ans;

int map[MAX];
int dist[MAX];
bool chk[MAX];

vector<vector<int>> adj;

void bfs(){
    queue<int> q;
    q.push(1);
    dist[1] = 0;
    chk[1] = true;
    
    while (!q.empty()) {
        int now = q.front();
        q.pop();
        
        for(int i=0; i<adj[now].size(); i++){
            int nxt = adj[now][i];
            
            if(!chk[nxt]){
                q.push(nxt);
                dist[nxt] = dist[now] + 1;
                chk[nxt] = true;
            }
        }
    }
    
    ans = 0;
    for(int i=1; i<=n; i++){
        if(dist[i]==1 || dist[i]==2) ans++;
    }
    cout << ans << "\n";
}

int main(int argc, const char * argv[]) {
    // cin,cout 속도향상
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    
    cin >> n;
    cin >> m;
    
    adj.resize(n+1);
    
    for(int i=0; i<m; i++){
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    
    bfs();
}