#include <bits/stdc++.h>
#define int long long
using namespace std;
using pii = pair<int, int>;
int n, k;
vector<vector<pii>> g;
vector<int> order, a, ans;
void dfs(int v, int p) {
order.push_back(v);
for (auto [u, uv] : g[v]) {
if (u != p) {
a.push_back(uv);
dfs(u, v);
}
}
}
void solve(int root) {
order.clear();
a.clear();
dfs(root, root);
vector<int> pref(n);
for (int i = 1; i < n; i++) {
pref[i] = pref[i - 1] + a[i - 1];
}
vector<int> z(n);
for (int i = 0; i < n - 1; i++) {
z[i] = lower_bound(pref.begin(), pref.end(), pref[i] + k + 1) - pref.begin() - 1;
}
vector<vector<int>> zT(n);
for (int i = 0; i < n - 1; i++) {
zT[z[i]].push_back(i);
}
vector<int> dp(n);
for (int i = 0; i < n; i++) {
dp[i] = 0;
for (int p : zT[i]) {
dp[i] += dp[p] + 1;
}
}
for (int i = 0; i < n; i++) {
ans[order[i]] += dp[i] * (n - i - 1);
}
}
signed main() {
ios_base::sync_with_stdio(0); cin.tie(0);
cin >> n >> k;
g.resize(n);
ans.resize(n);
for (int i = 0; i < n - 1; i++) {
int x, y, w;
cin >> x >> y >> w;
g[x].emplace_back(y, w);
g[y].emplace_back(x, w);
}
int root = 0;
for (int i = 0; i < n; i++) {
if (g[i].size() == 1) {
solve(i);
}
}
for (int i = 0; i < n; i++) {
cout << ans[i] << "\n";
}
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |