[Python] BOJ 14675 - 단절점과 단절선

https://www.acmicpc.net/problem/14675

14675번: 단절점과 단절선

# 트리문제이다.

# 우선 입력으로 주어지는 정보는 트리임이 보장된다고 했으므로 모든 간선은 단절선이 된다!

 

# 또한, 정점중에서 리프노드가 아닌 정점들은 모두 단절점이 된다.

   따라서 입력받을때 정점들을 graph배열에 넣고 확인하려는 정점 k의 graph[k]의 길이가 1이하라면

   리프노드이므로 no를 출력해주고 아니면 yes를 출력해준다.

 

import sys
input = sys.stdin.readline
N = int(input())
graph = [[] for _ in range(N+1)]
for _ in range(N-1):
    a, b = map(int,input().split())
    graph[a].append(b)
    graph[b].append(a)

q = int(input())
for _ in range(q):
    t, k = map(int,input().split())
    if t == 2:
        print("yes")
    else:
        if len(graph[k]) <= 1:
            print("no")
        else:
            print("yes")

(solved.ac 티어: 골드 5)