Codeforces Round 811 (Div. 3)G题题解

Path Prefixes

问题建模

给定一棵树，树的边上有两个权值，问除了根节点外的每个节点其到根节点最长前缀第二权值之和小于等于根节点到当前节点的第一权值之和的值。

问题分析

1.分析所求

求除根节点外每个节点的所求值，既需要有根节点到当前节点路径上第一权值的和，还要到当前节点的第二权值的各个前缀和。则可按树的dfs搜索方式，对于每个到达的每个节点累加上当前节点的第一权值，存储到当前节点的第二权值的前缀和，由于权值为正，则可对到当前节点的第二权值的前缀和数组使用二分，得到小于等于到当前节点第一个权值和的最长第二权值前缀和的长度。

代码

#include<bits/stdc++.h>

#define x first
#define y second
#define C(i) str[0][i]!=str[1][i]
using namespace std;
typedef unsigned long long ULL;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
const int N = 2e5 + 10, Mod = 998244353, P = 2048;
vector<int> e[N];
PII a[N];
LL sum1,sum2;
vector<LL> vt;
int n;
int ans[N];

void dfs(int u){
    sum1+=a[u].x;
    sum2+=a[u].y;
    vt.push_back(sum2);///到
    for(auto v:e[u])    dfs(v);
    int l=0,r=vt.size()-1;
    while(l<r){
        int mid=(l+r+1)>>1;
        if(vt[mid]<=sum1)   l=mid;
        else r=mid-1;
    }
    ans[u]=l;
    sum1-=a[u].x;
    sum2-=a[u].y;
    vt.pop_back();
}

void solve() {
    cin >>n;
    for(int i=1;i<=n;i++)   e[i].clear();
    for(int i=2;i<=n;i++){
        int p,at,b;
        scanf("%d %d %d",&p,&at,&b);
        e[p].push_back(i);
        a[i]={at,b};
    }
    sum1=0,sum2=0;
    dfs(1);
    for(int i=2;i<=n;i++){
        cout <<ans[i]<<" " ;
    }
    cout <<"\n";
}

int main() {
    int t = 1;
    cin >> t;
    while (t--) solve();
    return 0;
}