CodeForces 1714G Path Prefixes

原题链接：https://codeforces.com/contest/1714/problem/G

题意： 给你一棵有根的树。它包含 n 个顶点，从 1 到 n 编号。根是顶点 1。每条边都有两个权值A[i]，B[i]，从1开始到点 u （2 ≤ u ≤ n）路径中的所有边的权重A[i]之和 >= 从1开始到点 v （2 ≤ v ≤ u）路径中的所有边的权重B[i]之和，求出每一个点u对应v的最大深度

 

 

思路：首先所有的权重B[i]都是正的，所以前缀上的量是单调递增的。那么就可以使用二分搜索来寻找顶点的答案。那么剩下的就简单了，我们只需要快速找到路径前缀上的权重B[i]的和。如何记录呢？经过思考完全可以用深度优先搜索找到当前的路径的前缀和，并将当前路径的前缀总和存储在堆栈中:转到顶点，将总和添加到路径的末尾，以防万一在退出时将其删除。

AC代码：

#include <bits/stdc++.h>
typedef long long ll;

void solve() {
	int n;
	std::cin >> n;
	std::vector<std::vector<std::array<int, 3>>>a(n);
	for (int i = 1; i < n; i++) {
		int p, x, y;
		std::cin >> p >> x >> y;
		p--;
		a[p].push_back({i, x, y});
	}
	std::vector<int>ans(n);

	std::vector<ll>A(n), B(n), s;
	std::function<void(int)> dfs = [&](int u) {
		s.push_back(B[u]);
		if (u > 0) {
			ans[u] = std::upper_bound(s.begin(), s.end(), A[u]) - s.begin() - 1;
		}
		for (auto [v, x, y] : a[u]) {
			A[v] = A[u] + x;
			B[v] = B[u] + y;
			dfs(v);
		}
		s.pop_back();
	};
	dfs(0);
	for (int i = 1; i < n; i++) std::cout << ans[i] << " \n"[i==n-1];
}

int main() {
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(nullptr);
	int T;
	std::cin >> T;
	while (T--) {
		solve();
	}
    return 0;
}