[백준 05567] 결혼식 [Java]

문제

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

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

입력

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

출력

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

---

풀이

상근이의 친구와, 친구의 친구 수를 세주는 문제다.

 

입력을 받아 양방향 그래프를 생성해주고

bfs로 풀이했다. 만약 방문 횟수(친구 관계)가 2보다 크다면 탐색 대상에서 제외하자.

1 = 상근이

2 = 상근이 친구

3 = 상근이 친구의 친구

친구 관계( graph[prev][i] )이고, 초대하지 않은 친구( visited[i] == 0 )인 경우에만 연결된 다른 친구를 탐색하자.

초대한 친구 수( cnt )를 반환해 출력하면 된다.

---

소스코드

보기

---

출처

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