Link: http://codeforces.com/contest/1059/problem/E
首先预处理出从每个节点u开始往上一条路径内可以覆盖的最远的节点last[u],然后从树的每个叶子开始往上,对于每一个节点,贪心地希望从子树中拿到延伸上来最长的路径,DFS的时候,如果自己就是一条path的第一个节点(叶节点一定是第一个节点),返回这条路径可以到达的最上方的节点,然后每次判断一下自己是否是子树中可以到达的最远的,如果是,那么自己将同时也是一条新的path的第一个节点,更新ans,返回last[u]。否则返回子树中最远的。
*如何预处理出last[u]:维护dfs栈的前缀和,二分即可。
代码:
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
typedef long long ll;
typedef pair<int, ll> ppp;
const int maxn = 1e5 + 5;
vector<int> G[maxn];
ppp work[maxn];
int now, ans;
int n, L, last[maxn], dep[maxn];
ll w[maxn], S;
void Getlast(int u, int now) {
int l = max(0, now - L), r = now;
while (l < r) {
int mid = (l + r)>>1;
if (work[mid].se + S >= work[now].se) r = mid;
else l = mid + 1;
}
last[u] = work[l].fi;
}
void dfs(int u) {
now++;
work[now].fi = u;
work[now].se = work[now-1].se + w[u];
Getlast(u, now);
for (int i = 0; i < G[u].size(); i++) {
dep[G[u][i]] = dep[u] + 1;
dfs(G[u][i]);
}
now--;
}
int solve(int u) {
int mn = dep[u];
int ret = u;
for (int i = 0; i < G[u].size(); i++) {
int llll = solve(G[u][i]);
if (mn > dep[llll]) {
mn = dep[llll];
ret = llll;
}
}
if (ret != u) return ret;
ans++;
return last[u];
}
int p[maxn];
int main() {
scanf("%d%d%lld", &n, &L, &S);
for (int i = 1; i <= n; i++) {
scanf("%lld", &w[i]);
if (w[i] > S) return puts("-1"), 0;
}
for (int i = 2; i <= n; i++) scanf("%d", &p[i]), G[p[i]].push_back(i);
dep[1] = 1;
dfs(1);
solve(1);
printf("%d\n", ans);
return 0;
}