結果

問題 No.1488 Max Score of the Tree
ユーザー HIcoderHIcoder
提出日時 2023-07-21 21:31:58
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 93 ms / 2,000 ms
コード長 2,596 bytes
コンパイル時間 1,238 ms
コンパイル使用メモリ 99,836 KB
実行使用メモリ 82,432 KB
最終ジャッジ日時 2024-09-21 22:49:39
合計ジャッジ時間 3,393 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 86 ms
78,848 KB
testcase_01 AC 86 ms
75,776 KB
testcase_02 AC 88 ms
79,616 KB
testcase_03 AC 93 ms
82,432 KB
testcase_04 AC 90 ms
82,432 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 22 ms
22,912 KB
testcase_07 AC 51 ms
48,640 KB
testcase_08 AC 32 ms
34,816 KB
testcase_09 AC 28 ms
25,600 KB
testcase_10 AC 49 ms
49,920 KB
testcase_11 AC 92 ms
82,304 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 15 ms
14,592 KB
testcase_14 AC 42 ms
41,088 KB
testcase_15 AC 27 ms
26,752 KB
testcase_16 AC 6 ms
7,680 KB
testcase_17 AC 17 ms
17,920 KB
testcase_18 AC 57 ms
56,064 KB
testcase_19 AC 31 ms
32,512 KB
testcase_20 AC 14 ms
14,976 KB
testcase_21 AC 8 ms
9,216 KB
testcase_22 AC 26 ms
26,112 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 18 ms
21,248 KB
testcase_27 AC 5 ms
5,888 KB
testcase_28 AC 11 ms
11,648 KB
testcase_29 AC 14 ms
14,464 KB
testcase_30 AC 67 ms
66,560 KB
testcase_31 AC 90 ms
82,432 KB
権限があれば一括ダウンロードができます

ソースコード

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;
const ll INF=1LL<<60;

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> cnt(N);//cnt[i]=辺iをとおっていくことができる葉の数
    vector<int> depth(N);//0からどれだけ離れているか

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

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

        if(flag){
            //now自身が葉
            return 1;
        }

        return res;
        
    };

    dfs(dfs,0,-1);

    vector<pair<ll,ll>> load;
    for(int i=0;i<N-1;i++){ //i番目の辺
        
        load.emplace_back(C[i],C[i]*cnt[i]); //i番目の辺の重さとコスト
        edges.emplace_back(A[i],B[i],C[i],i);

    }

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

    vector<vector<ll>> dp(N,vector<ll>(K+1,-INF));
    dp[0][0]=0;
    for(int i=0;i<N-1;i++){
        for(int j=0;j<=K;j++){
            if(dp[i][j]==-INF) continue;

            dp[i+1][j]=max(dp[i+1][j],dp[i][j]);

            if(j+load[i].first<=K){
                dp[i+1][j+load[i].first]=max(dp[i+1][j+load[i].first],dp[i][j]+load[i].second);
            }
        }
    }

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

    ll add=0;

    for(int j=0;j<=K;j++){
        add=max(add,dp[N-1][j]);
    }
  

    ll ans=add+tot;

    cout<<ans<<endl;


}
0