結果

問題 No.417 チューリップバブル
ユーザー cormoran
提出日時 2016-12-20 18:21:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 73 ms / 2,000 ms
コード長 2,290 bytes
コンパイル時間 2,323 ms
コンパイル使用メモリ 186,992 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-14 11:32:45
合計ジャッジ時間 4,222 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using pii = pair<int,int>;
using ll = long long;
#define rep(i, j) for(int i=0; i < (int)(j); i++)
#define repeat(i, j, k) for(int i = (j); i < (int)(k); i++)
#define all(v) v.begin(),v.end()
#define debug(x) cerr << #x << " : " << x << endl

template<class T> bool set_min(T &a, const T &b) { return a > b  ? a = b, true : false; }
template<class T> bool set_max(T &a, const T &b) { return a < b  ? a = b, true : false; }
// vector
template<class T> istream& operator >> (istream &is , vector<T> &v) { for(T &a : v) is >> a; return is; }
template<class T> ostream& operator << (ostream &os , const vector<T> &v) { for(const T &t : v) os << "\t" << t; return os << endl; }
// pair
template<class T, class U> ostream& operator << (ostream &os , const pair<T, U> &v) { return os << "<" << v.first << ", " << v.second << ">"; }

const int INF = 1 << 30;
const ll INFL = 1LL << 60;


class Solver {
  public:
    int N, M;
    vector<int> U;
    vector<vector<pii>> E;

    vector<map<int, int>> memo;

    void rec(int now, int pre) {
        // cerr << "@" << now << " from " << pre << endl;
        memo[now][0] = U[now];
        
        for(auto e: E[now]) {
            if(e.first == pre) continue;
            rec(e.first, now);

            map<int, int> mp;
            for(auto np : memo[now]) {
                for(auto p : memo[e.first]) {                
                    int m = p.first + np.first + e.second * 2;
                    if(m <= M) set_max(mp[m], p.second + np.second);
                }
            }
            for(auto p : mp) {
                set_max(memo[now][p.first], p.second);
            }
        }
    }
    
    bool solve() {
        cin >> N >> M;
        U.resize(N); cin >> U;
        E.resize(N);
        rep(i, N - 1) {
            int a, b, c; cin >> a >> b >> c;
            E[a].push_back(make_pair(b, c));
            E[b].push_back(make_pair(a, c));
        }

        memo.resize(N);

        rec(0, -1);
        int ans = 0;
        for(auto p : memo[0]) {            
            set_max(ans, p.second);
        }

        cout << ans << endl;
        
        return 0;
    }
};

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    Solver s;
    s.solve();
    return 0;
}
0