[SWEA][Python] 5178 노드의 합

[SWEA][Python] 5178 노드의 합

리프노드 갯수랑 value 다 줘서 구현은 어렵지 않았다.

첨엔 문제만 읽고 Segment Tree 그리라는줄..

def prefix_tree(node): if N-M < node <= N: return tree[node] elif node > N: return 0 else: left = prefix_tree(node*2) right = prefix_tree(node*2 + 1) tree[node] = left + right return tree[node] for test in range(1, int(input())+1): N, M, L = map(int, input().split()) tree = [0] *(N+1) for _ in range(M): node_num, value = map(int, input().split()) tree[node_num] = value prefix_tree(1) print(f'#{test} {tree[L]}')

from http://devlibrary00108.tistory.com/320 by ccl(A) rewrite - 2021-08-24 01:26:18