백준14267 내리 갈굼

작성일 | In

문제 링크

  • http://icpc.me/14267

사용 알고리즘

  • DFS

시간복잡도

  • O(n + m)

풀이

트리의 각 노드마다 갈굼을 얼마나 받는지 미리 저장을 하고, dfs를 돌면서 부하들에게 전파해주면 됩니다.

전체 코드

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
#include <bits/stdc++.h>
using namespace std;

vector<int> g[101010];
int cost[101010];

void dfs(int v){
	for(auto u : g[v]){
		cost[u] += cost[v];
		dfs(u);
	}
}

int main(){
	ios_base::sync_with_stdio(0);
	int n, m; cin >> n >> m;
	for(int i=1; i<=n; i++){
		int t; cin >> t;
		if(i != 1) g[t].push_back(i);
	}

	while(m--){
		int a, b; cin >> a >> b;
		cost[a] += b;
	}

	dfs(1);
	for(int i=1; i<=n; i++) cout << cost[i] << " ";
}