13 ljedd2

[LJEDD2]

🔗 문제 링크

백준 플로이드 워샬 | BOJ 1058 친구

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

제한 조건

• N은 50보다 작거나 같은 자연수

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

✔️ 소요된 시간

2시간

✨ 수도 코드

지문 중 직접 친구이거나 k 친구를 거쳐 친구인지까지 확인하라는 내용이 포함되어 있어 얼마 전 복습했던 플로이드 워셜을 한번 떠올려볼 수 있었습니다.

디스코드에 해당 내용에 대한 노트를 정리해둔 것을 참고하였습니다.

(A의 2-친구) = (A의 친구) U (친구의 친구) 이므로
한 번 건너서 친구인 사람이 있는지 없는지 검사합니다.

플로이드 와샬 알고리즘 사용 후 거리 (visited) 값이 2 이하가 된다면 2-친구를 만족합니다.
2-친구의 수가 가장 많은 사람을 출력하면 됩니다.

최종 코드

import sys
input = sys.stdin.readline

n = int(input())
graph = [list(input().strip()) for _ in range(n)]
visited = [[0] * n for _ in range(n)]

for k in range(n):
    for i in range(n):
        for j in range(n):
            if i == j:
                continue

        # 2-친구인 경우
            if graph[i][j] == 'Y' or (graph[i][k] == 'Y' and graph[k][j] == 'Y'):
                visited [i][j] = 1

res = 0
for row in visited:
  res = max(res,sum(row))
print(res)

📚 새롭게 알게된 내용

[janghw0126]

플로이드 워셜 알고리즘은 아직 저에게 너무나두 익숙하지 않은 알고리즘인데 디스코드에 올려주신 알고리즘 노트 덕분에 개념이 확실하게 잡혔네요! 그치만 알고리즘은 이해해도 이 문제의 지문을 보았을 때 어느 포인트가 플로이드 워셜을 생각하게 만드는지 감을 못잡아서 결국 정은님의 수도코드를 참고하게 되었습니다🥺

다익스트라 알고리즘부터 한번 제대로 뿌수고 플로이드 워셜을 건드려봐야겠습니다😖

이 문제를 bfs로 해결하는 방법도 있습니다!

import sys
from collections import deque

def bfs(start):
    visited = [False] * n
    # 시작 노드와 깊이를 같이 큐에 저장함
    queue = deque([[start, 0]])
    visited[start] = True
    friends_count = 0

    while queue:
        current, depth = queue.popleft()

        # 깊이가 2 이상인 경우는 무시함
        if depth >= 2:  
            continue

        for neighbor in range(n):
            if not visited[neighbor] and graph[current][neighbor] == "Y":
		# 친구 수를 증가시킴
                friends_count += 1
		# 방문 처리를 함
                visited[neighbor] = True
		# 이웃과 깊이를 추가함
                queue.append([neighbor, depth + 1])

    return friends_count

# 노드 수를 입력함
n = int(sys.stdin.readline().strip())
# 그래프 정보를 입력함
graph = [list(sys.stdin.readline().strip()) for _ in range(n)]
max_friends = 0 

for i in range(n):
    # 각 노드별로 최대 친구 수를 갱신함
    max_friends = max(max_friends, bfs(i))

# 최대 친구 수를 출력함
print(max_friends)

다음 PR도 팟팅입니다🍀

[jung0115]

저도 알고리즘 노트 덕분에 헷갈리던 플로이드 워셜 알고리즘의 개념을 명확히 알게 된 것 같아요!
문제 내용을 보고 어떤 알고리즘을 사용해야 할지 바로 떠올리신 게 대단한 것 같습니다...! 🥹