백준 5567번: 결혼식 (JAVA)

 

문제 해석

 

친구의 친구까지만 결혼식에 초대할 수 있다. 결혼식에 초대 가능한 친구들은 총 몇명인지 출력한다.

상근이의 학번은 1이고 사람들은 각자의 학번을 가지고 있다.

 

알고리즘

 

친구 관계를 그래프로 구현하고 상근이를 기준으로 친구 또는 친구의 친구인 사람들은 상근이를 루트 노드로 생각했을 때 깊이가 1이거나 2인 사람들이다. 따라서 그래프 구현 후 BFS를 이용해 친구 관계를 탐색하는데 탐색 후 사람들의 깊이를 저장하고 깊이가 1 또는 2인 사람들의 수를 출력한다.

 

코드