1058 친구

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
2 초	128 MB	8002	3308	2758	42.634%

문제

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

A와 B가 친구면, B와 A도 친구이고, A와 A는 친구가 아니다.

입력

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

출력

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

예제 입력 1

3
NYY
YNY
YYN

예제 출력 1

2

예제 입력 2

3
NNN
NNN
NNN

예제 출력 2

0

예제 입력 3

5
NYNNN
YNYNN
NYNYN
NNYNY
NNNYN

예제 출력 3

4

---

풀이과정

(나의 2_친구) = (나의 친구) U (친구의 친구) 이므로

나의 친구와 친구의 친구를 하나의 배열에 넣고 중복을 제거한다. (set 이용)

2_친구 수 = 배열의 길이

---

코드

import sys

n = int(sys.stdin.readline())
input =  []

for i in range(n):
    input.append(list(sys.stdin.readline()))

graph = [[] for _ in range(n)]
for i in range(n):
    for j in range(n):
        if input[i][j] == 'Y':
            graph[i].append(j)

friend_count = [0] * (n)
for i in range(n):
    friend = set()
    for j in graph[i]:
        friend.add(j)
        for k in graph[j]:
            friend.add(k)
    if len(friend) == 0:
        friend_count[i] = 0
    else:
        friend_count[i] = len(friend)-1
        

print(max(friend_count))