結果

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

ソースコード

diff # 

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

    vector<int> A(N-1),B(N-1),C(N-1);
    for(int i=0;i<N-1;i++){
        
        cin>>A[i]>>B[i]>>C[i];
        A[i]--;
        B[i]--;

        G[A[i]].emplace_back(A[i],B[i],C[i],i);
        G[B[i]].emplace_back(B[i],A[i],C[i],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-1;i++){
        if(depth[A[i]]>depth[B[i]]) swap(A[i],B[i]);
        //depth[A[i]]<depth[B[i]];

        edges.emplace_back(A[i],B[i],C[i],i);

    }

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

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

    for(edge e:edges){

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

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


    }
    

    sort(dp.begin(),dp.end(),[&](const P&lhs, const P&rhs){
        if(lhs.first!= rhs.first){
            return lhs.first> rhs.first;
        }else{
            return lhs.second<rhs.second;
        }
    });

    //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+dp[i].second)<=K){
            //cout<<"dp[i].second="<<dp[i].second<<endl;
        

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

        }
    }


    //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