結果

問題 No.417 チューリップバブル
コンテスト
ユーザー Eliza_0x
提出日時 2018-05-19 17:30:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 953 ms / 2,000 ms
コード長 1,259 bytes
コンパイル時間 977 ms
コンパイル使用メモリ 89,992 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-28 14:25:47
合計ジャッジ時間 12,779 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <functional>
using namespace std; 

struct Edge {
    int to;
    int cost;
    Edge(int to, int cost): to(to), cost(cost) {}
};

int main() {
    int n, m; cin >> n >> m;
    vector<int> points(n);
    for(auto &&x: points) cin >> x;
    vector<vector<Edge>> edges(n);
    for (int i=0; i<n-1; i++) {
        int from, to, cost; cin >> from >> to >> cost;
        edges[from].push_back(Edge(to, cost));
        edges[to].push_back(Edge(from, cost));
    }
    vector<vector<int>> memo(n, vector<int>(m+1, 0));

    function<void(int, int)> dfs = [&](int cur, int prev) -> void {
        for (int i=0; i<m+1; i++) memo[cur][i] = points[cur];
        for (auto edge: edges[cur]) if (edge.to != prev) {
            dfs(edge.to, cur);
            for (int i=m; i>=0; i--) {
                for (int j=0; j<=i+edge.cost*2; j++) {
                    if (i-(j+edge.cost*2) >= 0)
                        memo[cur][i] = max(
                            memo[cur][i],
                            memo[cur][i-(j+edge.cost*2)] + memo[edge.to][j]);
                }
            }
        }
    };

    dfs(0, -1);
    cout << *max_element(memo[0].begin(), memo[0].end()) << endl;
}
0