結果
問題 |
No.1488 Max Score of the Tree
|
ユーザー |
![]() |
提出日時 | 2023-12-25 06:00:33 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 23 ms / 2,000 ms |
コード長 | 1,524 bytes |
コンパイル時間 | 1,322 ms |
コンパイル使用メモリ | 131,368 KB |
最終ジャッジ日時 | 2025-02-18 14:30:27 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
#include <cctype> #include <iostream> #include <vector> #include <algorithm> #include <cstring> #include <cstdlib> #include <cassert> #include <cmath> #include <ctime> #include <iomanip> #include <numeric> #include <stack> #include <queue> #include <map> #include <unordered_map> #include <set> #include <unordered_set> #include <bitset> #include <random> #include <functional> using namespace std; int N,K; int ch[100]; vector<pair<int,int>> G[100]; void dfs(int u,int p,vector<pair<long long,long long>> &W,long long &ans) { ch[u] = 0; if(p != -1 && G[u].size() == 1) ch[u] = 1; for(const auto &[v,w]:G[u]) if(v != p) { dfs(v,u,W,ans); ch[u] += ch[v]; ans += (long long)w*ch[v]; W.push_back(make_pair(w,(long long)ch[v]*w)); } } void solve() { cin >> N >> K; for(int i = 1;i < N;i++) { int u,v,w; cin >> u >> v >> w; u--; v--; G[u].push_back(make_pair(v,w)); G[v].push_back(make_pair(u,w)); } vector<pair<long long,long long>> W; long long ans = 0; dfs(0,-1,W,ans); vector<long long> dp(K+1,-(long long)1e18); dp[0] = 0; for(int i = 0;i < N-1;i++) { vector<long long> n_dp(K+1,-(long long)1e18); const auto &[w,v] = W[i]; for(int j = 0;j <= K;j++) { n_dp[j] = max(n_dp[j],dp[j]); if(j+w <= K) n_dp[j+w] = max(n_dp[j+w],dp[j] + v); } swap(dp,n_dp); } long long mx = 0; for(int i = 0;i <= K;i++) mx = max(mx,dp[i]); ans += mx; cout << ans << "\n"; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int tt = 1; /* cin >> tt; */ while(tt--) solve(); }