結果

問題 No.417 チューリップバブル
コンテスト
ユーザー nenuon
提出日時 2017-07-08 19:18:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,331 bytes
コンパイル時間 934 ms
コンパイル使用メモリ 93,268 KB
実行使用メモリ 7,936 KB
最終ジャッジ日時 2024-10-06 19:49:09
合計ジャッジ時間 2,220 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <map>
#include <cmath>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <vector>
#include <stdlib.h>
#include <stdio.h>
#include <bitset>
using namespace std;
#define FOR(I,A,B) for(int I = (A); I < (B); ++I)
typedef long long ll;

const int MMAX = 2222;
const int NMAX = 222;
ll N, M;
ll dp[NMAX][MMAX]; // 頂点番号, 時間, 0は確定バージョン
ll U[NMAX];
bool visited[NMAX];
vector<vector<ll> > G(NMAX);
ll C[NMAX][NMAX]; // かかる時間

void dfs(int v){
  visited[v] = true;
  dp[v][0] = U[v];
  for(auto &to : G[v]){
    if(!visited[to]){
      dfs(to);
      // vにたす
      for(int j = M; j >= 0; j--){
        if(dp[v][j] == 0) continue;
        FOR(i,0,M+1){
          if(dp[to][i] == 0) continue;
          int t = j + i + 2 * C[v][to];
          if(t <= M) dp[v][t] = max(dp[v][t], dp[v][j] + dp[to][i]);
        }
      }
    }
  }
  return;
}

int main(){
  cin >> N >> M;
  FOR(i,0,N) cin >> U[i];
  FOR(i,0,N-1){
    ll a, b, c;
    cin >> a >> b >> c;
    G[a].push_back(b);
    G[b].push_back(a);
    C[a][b] = c;
    C[b][a] = c;
  }
  FOR(i,0,NMAX) FOR(j,0,MMAX) dp[i][j] = 0;
  dfs(0);
  ll ans = 0;
  FOR(i,0,M+1) ans = max(ans, dp[0][i]);
  printf("%lld\n", ans);
  return 0;
}
0