[백준] 15681번:트리와 쿼리 (python 파이썬)

문제 www.acmicpc.net/problem/15681 15681번: 트리와 쿼리 트리의 정점의 수 N과 루트의 번호 R, 쿼리의 수 Q가 주어진다. (2 ≤ N ≤ 105, 1 ≤ R ≤ N, 1 ≤ Q ≤ 105) 이어 N-1줄에 걸쳐, U V의 형태로 트리에 속한 간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V) www.acmicpc.net 풀이 문제가 어려워 보이지만, 간단히 말하면 트리에서 마지막 입력받는 정점을 루트로하는 서브트리의 정점의 수를 출력하면 된다. dfs로 탐색하면서 각 정점에 대한 서브트리의 수를 count 배열에 담아준다. count 배열로 방문여부 표시도 같이 해줄 수 있다. 소스코드 import sys input = sys.stdin.readline sys..
Coding - Algo/python 2021. 3. 9. 18:07
1