BOJ: 1058

Posted on In

문제

지민이는 세계에서 가장 유명한 사람이 누구인지 궁금해졌다. 가장 유명한 사람을 구하는 방법은 각 사람의 2-친구를 구하면 된다. 어떤 사람 A가 또다른 사람 B의 2-친구가 되기 위해선, 두 사람이 친구이거나, A와 친구이고, B와 친구인 C가 존재해야 된다. 여기서 가장 유명한 사람은 2-친구의 수가 가장 많은 사람이다. 가장 유명한 사람의 2-친구의 수를 출력하는 프로그램을 작성하시오.
A와 B가 친구면, B와 A도 친구이고, A와 A는 친구가 아니다.

입력

첫째 줄에 사람의 수 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 각 사람이 친구이면 Y, 아니면 N이 주어진다.

출력

첫째 줄에 가장 유명한 사람의 2-친구의 수를 출력한다.

Solution

입력 구현

1
2
3
4
5
6
7
8
9
10
11
import sys
read = sys.stdin.readline

N = int(read())
g = [[] for _ in range(N)]

for i in range(N):
  s = read().rstrip()
  for j, k in enumerate(s):
    if k == 'Y':
      g[i].append(j)

BFS 구현

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from collections import deque

def friend(start):
  f = 0
  q = deque([(start, 0)])
  v = [False for _ in range(N)]
  v[start] = True
  while q:
    tmp = q.popleft()
    for i in g[tmp[0]]:
      if tmp[1] < 2 and not v[i]:
        v[i] = True
        q.append((i, tmp[1] + 1))
        f += 1
  return f
  • 큐 q에 현재 위치와 노드 횡단 수 (0-친구, 1-친구, 2-친구)를 튜플 자료형으로 입력하고 방문 여부를 기억할 v 변수 선언
  • BFS를 통해 그래프 내의 2-친구를 조회
    • 노드 횡단 수 tmp[1]이 2 미만
    • 방문을 하지 않은 경우 not v[i]
    • 2-친구의 수 f 1 증가

결과 출력

1
2
3
4
5
res = 0
for i in range(N):
  res = max(res, friend(i))

print(res)
  • 시작지점을 0부터 N - 1까지 선정하여 반복
  • 가장 큰 결과 출력