이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define ii pair <int, int>
#define F first
#define S second
using namespace std;
const int mod = 1e9 + 7;
const int N = 1e5 + 5;
int n, k, dem;
vector <ii> c[N];
long long ans[N];
long long dp[N];
void dfs(int u, int pa){
dp[u] = 0;
for (ii v : c[u]){
if (v.F == pa) continue;
dfs(v.F, u);
dp[u] = max(dp[u], dp[v.F] + v.S);
}
bool used = false;
for (ii v : c[u]){
if (v.F == pa) continue;
if (used or dp[v.F] + v.S < dp[u]){
++ dem;
ans[dem] = dp[v.F] + v.S;
}
else used = true;
}
}
long long solve_single_root(int r){
dem = 0;
dfs(r, 0);
++ dem;
ans[dem] = dp[r];
sort(ans + 1, ans + dem + 1);
long long res = 0;
for (int i = dem; i >= max(1, dem - k + 1); -- i) {
res += ans[i];
//if (r == 1) cout << ans[i] << ' ';
}
return res;
}
void solve(){
cin >> n >> k;
for (int i = 1; i <= n - 1; ++ i){
int u, v, w;
cin >> u >> v >> w;
c[u].push_back(ii(v, w));
c[v].push_back(ii(u, w));
}
for (int i = 1; i <= n; ++ i) cout << solve_single_root(i) << '\n';
}
int main(){
//freopen("code.inp", "r", stdin);
//freopen("code.out", "w", stdout);
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t = 1;
while (t --) solve();
return 0;
}
| # | 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... |