結果

問題 No.1488 Max Score of the Tree
ユーザー chocorusk
提出日時 2021-04-23 21:35:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 2,000 ms
コード長 1,446 bytes
コンパイル時間 3,707 ms
コンパイル使用メモリ 182,684 KB
最終ジャッジ日時 2025-01-20 23:38:11
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
int n;
int k;
vector<P> g[101];
int cnt[101];
int d[101];
vector<int> v, w, l;
void dfs(int x, int p){
    for(auto q:g[x]){
        int y=q.first;
        if(y==p) continue;
        d[y]=d[x]+q.second;
        dfs(y, x);
        cnt[x]+=cnt[y];
        v.push_back(q.second);
        w.push_back(q.second*cnt[y]);
    }
    if(cnt[x]==0) cnt[x]=1, l.push_back(x);
}
int dp[100010];
int main()
{
    cin>>n;
    cin>>k;
    for(int i=0; i<n-1; i++){
        int a, b, c; cin>>a>>b>>c;
        a--; b--;
        g[a].push_back({b, c});
        g[b].push_back({a, c});
    }
    dfs(0, -1);
    for(int i=0; i<n-1; i++){
        for(int j=k-v[i]; j>=0; j--){
            dp[j+v[i]]=max(dp[j+v[i]], dp[j]+w[i]);
        }
    }
    int ans=dp[k];
    //cout<<ans<<endl;
    for(auto x:l) if(cnt[x]==1) ans+=d[x];
    cout<<ans<<endl;
    return 0;
}
0