結果

問題 No.417 チューリップバブル
ユーザー motakine
提出日時 2020-05-13 16:30:15
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 313 ms / 2,000 ms
コード長 2,648 bytes
コンパイル時間 2,039 ms
コンパイル使用メモリ 186,460 KB
実行使用メモリ 9,856 KB
最終ジャッジ日時 2024-09-14 15:35:41
合計ジャッジ時間 6,840 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i=0; i<(int)(n); i++)
#define all(v) v.begin(), v.end()
#define PRINT(v) for (auto x : (v)) cout <<x <<" " ; cout <<endl;
using namespace std;
using ll = long long;
using Graph = vector<vector<int>>;
using mat = vector<vector<ll>>;
const ll MOD = 1000000007;
const ll INF = 10000000000000000;
const int inf = 1001001001;
vector<int> x4 = {0, 1, 0, -1}, x8 = {0, 1, 1, 1, 0, -1, -1, -1};
vector<int> y4 = {1, 0, -1, 0}, y8 = {1, 1, 0, -1, -1, -1, 0, 1};
template<class T> inline bool chmin(T& a, T b){if (a>b){a = b; return true;}return false;}
template<class T> inline bool chmax(T& a, T b){if (a<b){a = b; return true;}return false;}
template<class T> inline T powerM(T a,T b){if (b==0) return 1;
T tmp = powerM(a,b/2); if (b%2==0) return tmp*tmp%MOD; else return tmp*tmp%MOD*a%MOD; }
template<class T> inline T power(T a,T b,T m){ if (b==0) return 1;
  T tmp = power(a,b/2,m); if (b%2==0) return tmp*tmp%m; else return tmp*tmp%m*a%m; }
template<class T> inline T gcd(T a, T b){if (b==0) return a; return gcd(b, a%b);}
template<class T> inline T lcm(T a, T b){return a / gcd(a,b) * b;}
// ax+by=gcd(a,b)を解く
template<class T> inline T extgcd(T a,T b,T &x,T &y){if (b==0){x=1; y=0; return a;} T d=extgcd(b,a%b,y,x); y -= a/b*x; return d;}
void hey(){ cout <<"hey" <<endl; }

template<class T> struct edge { int to; T cost;};


int N,M;
vector<ll> U;
vector<vector<edge<int>>> G;
vector<vector<vector<ll>>> dp;


void input(){
  cin >>N >>M;
  U.assign(N, 0); rep(i, N) cin >>U[i];
  G.assign(N, vector<edge<int>>());
  rep(i, N-1){
    int a,b,c; cin >>a >>b >>c;
    G[a].push_back({b, c});
    G[b].push_back({a, c});
  }
  dp.assign(N, vector<vector<ll>>(2, vector<ll>(M+1)));
  // dp[i][j] := 頂点iからその部分木へ行って戻ってきて、時間jで取ってこれる税収の最大値
}

void dfs(int v, int p){
  for (auto pe : G[v]){
    int nv = pe.to;
    if (nv == p) continue;
    dfs(nv, v);
  }
  // 子の情報は全てdpに入っているとする
  dp[v][0][0] = U[v];
  dp[v][1][0] = U[v];
  // 時間0なら自分のやつは取って来れる
  for (auto pe : G[v]){
    int nv = pe.to;
    if (nv == p) continue;
    // dp[nv][j]を見ていく感じ
    // 後ろから見ていかないと更新が被りそう
    for (int j=M-pe.cost*2; j>=0; j--){
      int i = j + pe.cost * 2;
      for (int k=M-i; k>=0; k--){
        chmax(dp[v][1][k+j+pe.cost*2], dp[v][0][k] + dp[nv][0][j]);
      }
    }
    dp[v][0] = dp[v][1];
  }
}

int main() {
  input();
  dfs(0, -1);
  ll res = 0;
  for (int i=0; i<=M; i++) chmax(res, dp[0][0][i]);
  cout <<res <<endl;
}
0