CodeForces - 1099F Cookies

题意：

给定一棵有 $n$ 个结点的树，$A$ 起点在根位置，$A$ 先移动，每次往下走一层，$B$ 每次可选择切断 $A$ 当前所在结点的一条连向子结点的边。$A$ 走过一条边时会花费相应代价，每个结点上有 $x_i$ 个曲奇，$A$ 可以选择吃其中任意多个，每个花费 $t_i$ 时间。$A$ 可以选择在某个结点的时候结束游戏，并利用剩余时间走回根结点。若 $A$ 有总时间 $T$， $A$，$B$ 都采取最优策略，问 $A$ 能吃到的最多的曲奇数量。$(n\le 10^5)$

链接：

https://codeforces.com/contest/1099/problem/F

解题思路：

类似博弈，若能得到在每个结点返回时候能吃到的最多的曲奇数量，则至下而上维护每个结点走到子结点的次大值即可得到根结点处出发的答案。每个结点处能吃到的最多的曲奇，可以贪心选择路径上的花费时间最少的那部分曲奇，则在权值线段树二分即可。

参考代码：

#include<bits/stdc++.h>
 
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define sz(a) ((int)a.size())
#define pb push_back
#define lson (rt << 1)
#define rson (rt << 1 | 1)
#define gmid (l + r >> 1)
const int maxn = 1e6 + 5;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
 
vector<pii> G[maxn];
ll xi[maxn], ti[maxn], num[maxn << 2], sum[maxn << 2];
ll dis[maxn], dp[maxn];
int n; ll t;
 
void pushUp(int rt){
 
    sum[rt] = sum[lson] + sum[rson];
    num[rt] = num[lson] + num[rson];
}
 
void update(int l, int r, int rt, int pos, int val){
 
    if(l == r){
 
        num[rt] += val;
        sum[rt] += pos * 1ll * val;
        return;
    }
    int mid = gmid;
    if(pos <= mid) update(l, mid, lson, pos, val);
    else update(mid + 1, r, rson, pos, val);
    pushUp(rt);
}
 
ll query(int l, int r, int rt, ll val){
 
    if(l == r) return min(val / l, num[rt]);
    int mid = gmid;
    if(val < sum[lson]) return query(l, mid, lson, val);
    else return num[lson] + query(mid + 1, r, rson, val - sum[lson]);
}
 
void dfs1(int u){
 
    if(dis[u] > t) { dp[u] = -1; return; }
    update(1, 1e6, 1, ti[u], xi[u]);
    dp[u] = query(1, 1e6, 1, t - dis[u]);
    for(int i = 0; i < sz(G[u]); ++i){
 
        int v = G[u][i].second, w = G[u][i].first;
        dis[v] = dis[u] + w;
        dfs1(v);
    }
    update(1, 1e6, 1, ti[u], -xi[u]);
}
 
void dfs2(int u){
 
    ll mx = -1, mx2 = -1;
    for(int i = 0; i < sz(G[u]); ++i){
 
        int v = G[u][i].second;
        if(dp[v] == -1) continue;
        dfs2(v);
        if(dp[v] > mx){
 
            mx2 = mx;
            mx = dp[v];
        }
        else if(dp[v] > mx2){
 
            mx2 = dp[v];
        }
    }
    // cout << u << " xx "<< mn2 << endl;
    dp[u] = max(dp[u], mx2);
    if(u == 1) dp[u] = max(dp[u], mx);
}
 
int main(){
 
    scanf("%d%lld", &n, &t);
    for(int i = 1; i <= n; ++i) scanf("%lld", &xi[i]);
    for(int i = 1; i <= n; ++i) scanf("%lld", &ti[i]);
    for(int i = 2; i <= n; ++i){
 
        int par, w; scanf("%d%d", &par, &w);
        G[par].pb({w * 2, i});
    }
    dfs1(1);
    // for(int i = 1; i <= n; ++i) cout << i << " No.1  " << dp[i] << endl;
    dfs2(1);
    // for(int i = 1; i <= n; ++i) cout << i << " No.2  " << dp[i] << endl;
    cout << dp[1] << endl;
    return 0;
}