Submission #1299064

#TimeUsernameProblemLanguageResultExecution timeMemory
1299064asli_bgPaths (RMI21_paths)C++20
19 / 100
5 ms3656 KiB
#include<bits/stdc++.h>
using namespace std;

#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;

#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;

typedef tree<pii,null_type,less<pii>,rb_tree_tag,tree_order_statistics_node_update> oset;

#define all(x) x.begin(),x.end()
#define cont(x) for(auto el:x) cout<<el<<' ';cout<<endl;
#define contp(x) for(auto el:x) cout<<el.fi<<'-'<<el.se<<' ';cout<<endl;

#define DEBUG(x) cout<<#x sp x<<endl;

#define carp(a,b) ((a%MOD)*(b%MOD))%MOD
#define topla(a,b) ((a%MOD)+(b%MOD))%MOD

#define mid (l+r)/2

const int MAXN=2e3+5;
const int MOD=1e9+7;

int n,k;
vii adj[MAXN];

deque<int> mx[MAXN], dp[MAXN]; //hepsi büyükten küçüğe sorted

deque<int> merge(deque<int> a,deque<int> b,int cost){
    deque<int> c;

    if(cost){
        if(b.empty()) b.pb(cost);
        else b[0]+=cost;
    }

    int pt1=0;
    int pt2=0;
    int sz1=a.size();
    int sz2=b.size();

    while(pt1<sz1 or pt2<sz2){
        if(pt2>=sz2 or (pt1<sz1 and a[pt1]>=b[pt2])){
            c.pb(a[pt1++]);
        }
        else{
            c.pb(b[pt2++]);
        }

        if(c.size()==k) break;
    }

    return c;
}

void dfs(int nd,int ata){
    for(auto [kom,cost]:adj[nd]){
        if(kom==ata) continue;
        dfs(kom,nd);

        mx[nd]=merge(mx[nd],mx[kom],cost);
    }
}

void dfs2(int nd,int ata){
    vii vec;

    for(auto [kom,cost]:adj[nd]){
        if(kom==ata) continue;
        vec.pb({kom,cost});
    }

    int sz=vec.size();

    deque<int> cur;

    //pref hesapla
    FOR(i,sz){
        int kom=vec[i].fi;
        int cost=vec[i].se;

        dp[kom]=merge(dp[kom],cur,0);

        cur=merge(cur,mx[kom],cost);
    }

    cur.clear();

    //suf hesapla
    for(int i=sz-1;i>=0;i--){
        int kom=vec[i].fi;
        int cost=vec[i].se;

        dp[kom]=merge(dp[kom],cur,0);

        cur=merge(cur,mx[kom],cost);
    }

    FOR(i,sz){
        int kom=vec[i].fi;
        int cost=vec[i].se;

        dp[kom]=merge(dp[kom],dp[nd],0);

        if(!dp[kom].empty()) dp[kom][0]+=cost;
    }

    for(auto el:vec) dfs2(el.fi,nd);
}

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});
    }

    dfs(1,-1);
    dfs2(1,-1);

    deque<int> vec;
    FORE(i,1,n+1){
        vec=merge(dp[i],mx[i],0);

        int ans=0;
        for(auto el:vec) ans+=el;
        cout<<ans<<endl;
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...