[백준] 5567번: 결혼식 - C++

문제

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

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

 

입력

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

 

출력

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

---

그래프 탐색을 이용해서 푸는 문제. BFS를 이용해서 풀었다.

최대한 내 힘으로, 내 방식으로 풀어보려고 아무것도 안보고 풀었는데 효율적이지도 않고 예쁘지도 않은 코드라 마음에도 안들고 당연히 이거보다 더 간단하게 푸는 방법도 있을것이다...

 

친구의 친구들까지만 결혼식에 초대할 수 있기 때문에, 그래프에서 1 기준으로 깊이가 2 이상 되는 친구들부터는 생각하지 않는다. 따라서 1에서부터의 깊이를 저장하는 배열을 하나 더 만들어 깊이가 3 이상이 되었을 경우 탐색을 멈추고 지금까지 센 친구의 수를 출력시킨다.

 

처음에 깊이 부터 생각해서 DFS로 풀었는데 그러면 걸리는 예제가 있는 것 같더라...

 

코드

#include <cstdio>
#include <queue>
#define MAX 501

using namespace std;

int friends[MAX][MAX] = { 0 };
bool visited[MAX] = { false };
int depth[MAX] = { 0 };

int count = 0;
int n, m;
void BFS(int num);

int main(void) {
    scanf("%d %d", &n, &m);

    for(int i = 0; i < m; i++) {
        int a, b;
        scanf("%d %d", &a, &b);
        friends[a][b] = friends[b][a] = 1;
    }

    BFS(1);
    printf("%d\n", count);
    return 0;
}

void BFS(int num) {
    queue<int> q;
    q.push(num);
    visited[num] = true;

    while(!q.empty()) {
        int vertex = q.front();
        q.pop();
        
        for(int i = 1; i <= n; i++) {
            if(visited[i] || friends[vertex][i] == 0)
                continue;
            q.push(i);
            visited[i] = true;
            depth[i] = depth[vertex] + 1;
            if(depth[i] > 2) return;

            count++;
        }
    }
}