글 읽기 - 정답은 맞혔고 이해도 했지만, 처음 코드가 왜 틀린지 이해가 가지 않습니다.

hosahn 2년 전 0

보시는것처럼 저는 수 배열의 index를 노드번호로, 배열의 값을 부모 노드번호로 해서 계속 갱신해주는 방식으로 Union Find를 구현하였는데요, 답은 맞게 나오는 것 같은데도 자꾸 틀리네요, 혹시 이유를 아시는 고수님 계실까요?? ㅠㅠ

import sys
input = sys.stdin.readline

def union_find(graph_parent, cur_num, target_num) :
    if graph_parent[cur_num] == cur_num and cur_num != target_num:
        return "NO"
    elif graph_parent[cur_num] == target_num :
        return "YES"

    else :
        union_find(graph_parent, graph_parent[cur_num], target_num)
    return

n, m = map(int, input().split(" "))
graph_parent = [i for i in range(n + 1)]

for i in range(m) :
    mode, groupA, groupB = map(int, input().split(" "))
    if mode == 1 :
        if groupA == groupB :
            print("YES")
            continue
        else :
            value = union_find(graph_parent, groupA, groupB)
            print(value)
    elif mode == 0 :
            if graph_parent[groupA] > graph_parent[groupB] :
                graph_parent[groupA] = graph_parent[groupB]
            elif graph_parent[groupA] < graph_parent[groupB] :
                graph_parent[groupB] = graph_parent[groupA]

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

정답은 맞혔고 이해도 했지만, 처음 코드가 왜 틀린지 이해가 가지 않습니다.

hosahn 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말