結果

問題 No.417 チューリップバブル
コンテスト
ユーザー izuru_matsuura
提出日時 2016-12-20 18:29:10
言語 D
(dmd 2.109.1)
結果
WA  
実行時間 -
コード長 1,997 bytes
コンパイル時間 1,739 ms
コンパイル使用メモリ 145,152 KB
実行使用メモリ 5,760 KB
最終ジャッジ日時 2024-06-12 06:00:49
合計ジャッジ時間 8,110 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 9 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm;
import std.array;
import std.ascii;
import std.container;
import std.conv;
import std.math;
import std.numeric;
import std.range;
import std.stdio;
import std.string;
import std.typecons;
void log(A...)(A arg) { stderr.writeln(arg); }
int size(T)(in T s) { return cast(int)s.length; }

struct Edge {
    int from, to, cost;
}

void main() {
    int N, M; readf("%s %s\n", &N, &M);
    auto U = new int[N];
    foreach (i; 0 .. N) {
        U[i] = readln.chomp.to!int;
    }
    auto G = new Edge[][N];
    auto D = new int[][](N, N);
    foreach (i; 0 .. N) foreach (j; 0 .. N) D[i][j] = int.max / 4;
    foreach (i; 0 .. N - 1) {
        int s, t, c; readf("%s %s %s\n", &s, &t, &c);
        G[s] ~= Edge(s, t, c);
        G[t] ~= Edge(t, s, c);
        D[s][t] = D[t][s] = c;
    }

    auto Vi = new int[N];
    auto Vp = new int[N];
    void dfs(int p, int v, ref int i) {
        //log([p, v, i]);
        Vi[i] = v;
        Vp[i] = p;
        i++;
        foreach (e; G[v]) {
            if (e.to == p) continue;
            dfs(v, e.to, i);
        }
    }
    int _ = 0; dfs(-1, 0, _);
    //log("Vi: ", Vi);
    //log("Vp: ", Vp);

    auto dp = new int[][][](2, N + 1, M + 1);
    dp[0][0][0] = U[0];
    for (int i = 1; i < N; i++) {
        for (int t = 0; t <= M; t++) {
            int p = Vp[i];
            int d = D[Vi[i]][p];
            dp[i % 2][Vi[i]][t] = max(dp[i % 2][Vi[i]][t], dp[(i - 1) % 2][Vi[i]][t]);
            if (t - 2 * d >= 0) dp[i % 2][Vi[i]][t] = dp[(i - 1) % 2][p][t - 2 * d];
            for (int j = 0; j < i; j++) {
                dp[i % 2][Vi[j]][t] = max(dp[i % 2][Vi[j]][t], dp[(i - 1) % 2][Vi[j]][t]);
                if (t - 1 >= 0) dp[i % 2][Vi[j]][t] = max(dp[i % 2][Vi[j]][t], dp[i % 2][Vi[j]][t - 1]);
                if (t - 2 * d >= 0) 
                    dp[i % 2][Vi[j]][t] = max(dp[i % 2][Vi[j]][t], dp[(i - 1) % 2][Vi[j]][t - 2 * d] + U[Vi[i]]);
            }
        }
    }
    writeln(dp[(N - 1) % 2][0][M]);


}
0