結果

問題 No.417 チューリップバブル
ユーザー momoyuu
提出日時 2023-03-23 16:54:04
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 502 ms / 2,000 ms
コード長 977 bytes
コンパイル時間 2,969 ms
コンパイル使用メモリ 250,660 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-18 15:31:06
合計ジャッジ時間 9,678 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll = long long;

int n,m;
vector<pair<int,int> > g[210];
ll u[210];

vector<ll> dfs(int ni,int p){
    vector<ll> dp(m+1,u[ni]);
    for(auto&itr:g[ni])if(itr.first!=p){
        vector<ll> nxt = dfs(itr.first,ni);
        vector<ll> dp1(m+1,0);
        for(int i = 0;i<=m;i++) dp1[i] = dp[i];
        for(int i = 0;i<=m;i++){
            for(int j = 0;j<=i;j++){
                int rest = i - j - 2 * itr.second;
                if(rest<0) break;
                dp1[i] = max(dp1[i],max(dp[i],dp[j]+nxt[rest]));
            }
        }
        for(int i = 1;i<=m;i++) dp1[i] = max(dp1[i],dp1[i-1]);
        swap(dp,dp1);
    }
    return dp;
}


int main(){
    cin>>n>>m;
    for(int i = 0;i<n;i++) cin>>u[i];
    for(int i = 1;i<n;i++){
        int u,v,c;
        cin>>u>>v>>c;
        g[u].push_back(make_pair(v,c));
        g[v].push_back(make_pair(u,c));
    }
    vector<ll> ans = dfs(0,-1);
    cout<<ans[m]<<endl;
}
0