-
Notifications
You must be signed in to change notification settings - Fork 651
Expand file tree
/
Copy path1949.cpp
More file actions
43 lines (36 loc) · 766 Bytes
/
Copy path1949.cpp
File metadata and controls
43 lines (36 loc) · 766 Bytes
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
32
33
34
35
36
37
38
39
40
41
42
43
// Authored by : BaaaaaaaaaaarkingDog
// Co-authored by : -
// http://boj.kr/6c03e03f2b254b3eb419eea37fe5caff
#include <bits/stdc++.h>
using namespace std;
int n;
int a[10005];
int d1[10005];
int d2[10005];
vector<int> adj[10005];
// d1[cur]와 d2[cur]를 계산하는 함수
void dfs(int cur, int par){
d1[cur] = a[cur];
d2[cur] = 0;
for(int nxt : adj[cur]){
if(par == nxt) continue;
dfs(nxt, cur);
d1[cur] += d2[nxt];
d2[cur] += max(d1[nxt], d2[nxt]);
}
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
cin >> n;
for(int i = 1; i <= n; i++)
cin >> a[i];
for(int i = 0; i < n-1; i++){
int u, v;
cin >> u >> v;
adj[u].push_back(v);
adj[v].push_back(u);
}
dfs(1, 0);
cout << max(d1[1], d2[1]);
}