結果

問題 No.1488 Max Score of the Tree
コンテスト
ユーザー HIcoder
提出日時 2023-07-21 20:46:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,829 bytes
コンパイル時間 883 ms
コンパイル使用メモリ 100,200 KB
最終ジャッジ日時 2025-02-15 16:13:29
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<iostream>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<algorithm>
#include<tuple>
using namespace std;
typedef pair<int,int> P;
typedef long long ll;

struct edge{
    int from;
    int to;
    ll cost;
    int id;
    edge(int from_,int to_,ll cost_,int id_):from(from_),to(to_),cost(cost_),id(id_){};
};

int main(){
    int N,K;
    cin>>N>>K;
    vector<vector<edge>> G(N);

    vector<edge> edges;

    for(int i=0;i<N-1;i++){
        int a,b;
        ll c;
        cin>>a>>b>>c;
        a--;b--;
        //edges[i]=edge(a,b,c,i);
        //edges[i]=(edge){a,b,c,i};
        edges.emplace_back(a,b,c,i);


        G[a].emplace_back(a,b,c,i);
        G[b].emplace_back(b,a,c,i);
    }

    vector<int> num(N);//num[i]=iの部分木にある葉の数
    vector<int> depth(N);//0からどれだけ離れているか

    //部分木の葉の数を数える
    auto dfs=[&](auto f,int now,int pre,int d)->int{
        bool flag=true;
        int res=0;

        for(edge e:G[now]){
            if(e.to==pre) continue;
            flag=false;
            res+=f(f,e.to,now,d+1); //e.toの部分木にある葉の数
        }
        depth[now]=d;

        if(flag){
            //now自身が葉
            num[now]=1;
            return 1;
        }

        return num[now]=res;
        
    };

    dfs(dfs,0,-1,0);

    /*
    for(int i=0;i<N;i++){
        cout<<"depth["<<i<<"]="<<depth[i]<<endl;

        cout<<"num["<<i<<"]="<<num[i]<<endl;
    }
    */
    
    vector<pair<ll,int>> dp;
 

    for(edge& e:edges){
        if(depth[e.from]>depth[e.to]) swap(e.from,e.to);

        //depth[e.from]<depth[e.to]

        //e.toの部分木にある頂点数
        dp.emplace_back(1LL*num[e.to]*e.cost,e.id);


    }
    

    sort(dp.begin(),dp.end(),[&](const P&lhs, const P&rhs){
        return lhs.first> rhs.first;
    });

    //cout<<"te"<<endl;


    ll add=0;
    ll len=0;
    for(int i=0;i<dp.size();i++){
        //cout<<"dp[i].first="<<dp[i].first<<endl;
        //cout<<"dp[i].second="<<dp[i].second<<endl;
       
        if((len+edges[dp[i].second].cost)<=K){
            //cout<<"dp[i].second="<<dp[i].second<<endl;
        

            add += 1LL*dp[i].first;
            len+=edges[dp[i].second].cost;

        }
    }


    //cout<<"te"<<endl;

    ll tot=0;

    auto dfs2=[&](auto f,int now,int pre,ll depth)->void{
        bool flag=true;
        int res=0;
        for(edge e:G[now]){
            if(e.to==pre) continue;
            flag=false;
            f(f,e.to,now,depth+e.cost);
        }

        if(flag){
            //nowが根の場合のみ　足される
            tot+=depth;
        }
        return;
    };

    dfs2(dfs2,0,-1,0);

    
    //cout<<"te"<<endl;
    //cout<<"tot="<<tot<<endl;
    

    ll ans=add+tot;

    cout<<ans<<endl;


}
0