結果
問題 | No.417 チューリップバブル |
ユーザー |
|
提出日時 | 2023-01-21 15:19:28 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 4 ms / 2,000 ms |
コード長 | 1,236 bytes |
コンパイル時間 | 1,886 ms |
コンパイル使用メモリ | 179,464 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-06-23 23:15:40 |
合計ジャッジ時間 | 3,247 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 40 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;int main(){ios::sync_with_stdio(false);cin.tie(0);int n, m, u, v, c;cin >> n >> m;vector<int> a(n);vector<vector<pair<int,int>>> g(n);vector<vector<int>> dp(n);for(auto &&v:a)cin >> v;for(int i = 1; i < n; i++){cin >> u >> v >> c;g[u].emplace_back(v, c);g[v].emplace_back(u, c);}m /= 2;function<void(int,int,int)> dfs = [&](int v, int p, int d){vector<int> &temp = dp[v];temp = {a[v]};int u, c;for(auto &&pa:g[v]){tie(u, c) = pa;if(u == p || c > d)continue;dfs(u, v, d - c);vector<int> &temp2 = dp[u];vector<int> ndp(min(d + 1, int(temp.size() + temp2.size()) + c - 1));for(int i = 0; i < temp.size(); i++){ndp[i] = max(ndp[i], temp[i]);for(int j = 0; j < temp2.size() && i + j + c <= d; j++){ndp[i + j + c] = max(ndp[i + j + c], temp[i] + temp2[j]);}}swap(temp, ndp);}};dfs(0, -1, m);cout << *max_element(dp[0].begin(), dp[0].end()) << '\n';}