#893671 (C++17) No.1488 Max Score of the Tree

• 問題: No.1488 Max Score of the Tree
• ユーザー: HIcoder
• 提出日時: 2023-07-21 21:31:58
• 言語: C++17 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 88 ms / 2,000 ms
• コード長: 2,596 bytes
• コンパイル時間: 1,147 ms
• コンパイル使用メモリ: 100,080 KB
• 最終ジャッジ日時: 2025-02-15 16:24:02
• ジャッジサーバーID （参考情報）: judge5 / judge1

ファイルパターン	結果
sample	AC * 3
other	AC * 29

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#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;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX