(Python/🥈2)백준리즘알고 1058번: 친구

 

문제 바로가기 

친구

문제:

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

입력:

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

출력:

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

 

풀이:

import sys

sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

n = int(input())
graph = [list(input()) for _ in range(n)]

friends = [[0] * n for _ in range(n)]

for k in range(n):
    for x in range(n):
        for y in range(n):
            if x != y:
                if graph[x][y] == "Y" or (graph[x][k] == "Y" and graph[k][y] == "Y"):
                    friends[x][y] = 1


print(max([sum(i) for i in friends]))

 

 

내가 푼 방법은 플로이드-워셜 알고리즘이다.

 

이 문제에서 

두 사람이 친구이거나, A와 친구이고, B와 친구인 C  이 부분을 포인트로 잡았는데

즉 A->C->B  가 된다는 것인데 그러면 플로이드-워셜을 사용하면 된다고 생각했다.