[python] 백준 1976번 - 여행 가자

1976번: 여행 가자 (acmicpc.net)

1976번: 여행 가자

 

import sys
input = sys.stdin.readline

n = int(input())
m = int(input())
node = [list(map(int, input().split()))for _ in range(n)]
parent = [i for i in range(n+1)]
arr = list(map(int, input().split()))

def find(x):
    if parent[x] != x:
        parent[x] = find(parent[x])
    return parent[x]

def union(x, y):
    x, y = find(x), find(y)
    if x > y:
        parent[x] = y
    else:
        parent[y] = x

for i in range(n):
    for j in range(n):
        if node[i][j]:
            union(i+1, j+1)

ans = "YES"
for i in range(1, m):
    if parent[arr[i]] != parent[arr[0]]:
        ans = "NO"
        break

print(ans)

오늘의 추천곡이에용~~

덕분에 빵딩이 흔들면서 풀었네요~

신나는 노래와

애플워치를 산 기분이 합쳐지니까

천국이네요 ㅎㅎ 애플 최고