-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbandit.cpp
49 lines (38 loc) · 872 Bytes
/
bandit.cpp
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
44
45
46
47
48
49
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pi = pair<ll, ll>;
using vi = vector<ll>;
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
ll n;
cin >> n;
vector<set<ll>> ed(n, set<ll>({}));
for (ll i = 1; i < n; i++) {
ll p;
cin >> p;
p--;
ed[p].insert(i);
}
vi a(n);
for (ll i = 0; i < n; i++) {
cin >> a[i];
}
ll max_ans = 0;
function<pi(ll)> dfs = [&](ll v) {
ll sum = a[v];
ll sz = 0;
for (ll child : ed[v]) {
pi ar = dfs(child);
sum += ar.first;
sz += ar.second;
}
if (sz == 0) sz = 1;
max_ans = max(max_ans, (ll)ceil((double)sum / (double)sz));
return make_pair(sum, sz);
};
dfs(0);
cout << max_ans << endl;
return 0;
}