티스토리 뷰
728x90
문제
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.setrecursionlimit(10**9)
def countPoint(x):
count[x]=1
for i in tree[x]:
if not count[i]:
countPoint(i)
count[x] += count[i]
n, r, q = map(int, input().split())
tree = [[] for _ in range(n+1)]
count = [0]*(n+1)
for i in range(n-1):
a, b = map(int, input().split())
tree[a].append(b)
tree[b].append(a)
countPoint(r)
for i in range(q):
u = int(input())
print(count[u])
반응형
'Coding - Algo > python' 카테고리의 다른 글
[백준] 11723번:집합 (python 파이썬) (0) | 2021.03.15 |
---|---|
[백준] 9095번:1, 2, 3 더하기 (python 파이썬) (0) | 2021.03.12 |
[백준] 2565번:전깃줄 (python 파이썬) (1) | 2021.03.08 |
[백준] 11726번:2xn 타일링 (python 파이썬) (0) | 2021.03.06 |
[백준] 1003번:피보나치 함수 (python 파이썬) (0) | 2021.03.05 |
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- yoloV3
- poker swea
- 백준 dp 문제
- SSAFY
- 메뉴리뉴얼 풀이
- 백준
- swea 타일링
- 1699 자바
- 우분투
- ubuntu
- swea 1240
- 파이썬
- 파이썬 풀이
- 백준 풀이
- 프로그래머스 파이썬
- 1240 자바
- 프로그래머스 자바
- swea 1240 자바
- 프로그래머스 더 맵게
- 더 맵게
- swea 4070 타일링
- 3996 자바
- SWEA
- 타일링 자바
- 프로그래머스
- swea 타일링 자바
- 백준 17144
- 삼성청년SW아카데미
- union-find
- 백준파이썬
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | ||||||
2 | 3 | 4 | 5 | 6 | 7 | 8 |
9 | 10 | 11 | 12 | 13 | 14 | 15 |
16 | 17 | 18 | 19 | 20 | 21 | 22 |
23 | 24 | 25 | 26 | 27 | 28 | 29 |
30 | 31 |
글 보관함