#include<bits/stdc++.h>
using namespace std;
#define int long long
#define FOR(i,a) for(int i=0;i<(a);i++)
#define FORE(i,a,b) for(int i=(a);i<(b);i++)
#define sp <<' '<<
#define fi first
#define se second
#define pb push_back
typedef vector<int> vi;
typedef pair<int,int> pii;
typedef vector<pii> vii;
typedef vector<bool> vb;
#define all(x) x.begin(),x.end()
const int MAXN=2e3+5;
const int INF=1e18;
const int MOD=1e9+7;
int n,k;
vii adj[MAXN];
multiset<int> tut[MAXN];
int top[MAXN];
void merge(multiset<int>& a,multiset<int>& b,int nd,int kom){
if(a.size()<b.size()){
a.swap(b);
swap(top[nd],top[kom]);
}
int sz=a.size();
for(auto el:b){
a.insert(el);
top[nd]+=el;
sz++;
if(sz>k){
auto el2=*a.begin();
top[nd]-=el2;
a.erase(a.find(el2));
sz--;
}
}
}
void dfs(int nd,int ata){
tut[nd].clear();
top[nd]=0;
for(auto [kom,cost]:adj[nd]){
if(kom==ata) continue;
dfs(kom,nd);
int el=0;
if(!tut[kom].empty()){
el=*tut[kom].rbegin();
tut[kom].erase(tut[kom].find(el));
top[kom]-=el;
}
tut[kom].insert(el+cost);
top[kom]+=el+cost;
merge(tut[nd],tut[kom],nd,kom);
}
}
signed main(){
ios_base::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin>>n>>k;
FOR(i,n-1){
int a,b,c;
cin>>a>>b>>c;
adj[a].pb({b,c});
adj[b].pb({a,c});
}
FORE(i,1,n+1){
//solve for all roots
dfs(i,-1);
cout<<top[i]<<endl;
}
}
| # | 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... |