Codeforces 1059E. Split the Tree

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;
}