結果

問題 No.1488 Max Score of the Tree
ユーザー daut-dlang
提出日時 2021-04-23 21:46:12
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 2,698 bytes
コンパイル時間 833 ms
コンパイル使用メモリ 121,572 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-22 11:12:44
合計ジャッジ時間 2,251 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import std.stdio, std.conv, std.functional, std.string;
import std.algorithm, std.array, std.container, std.range, std.typecons;
import std.bigint, std.numeric, std.math, std.random;
import core.bitop;
string FMT_F = "%.10f";
static File _f;
void file_io(string fn) { _f = File(fn, "r"); }
static string[] s_rd;
T _RD(T = long)() { while(!s_rd.length) s_rd = readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T _RD(T = long)(File f) { while(!s_rd.length) s_rd = f.readln.chomp.split; string res = s_rd[0]; s_rd.popFront; return res.to!T; }
T[] _RDA(T = long)(T fix = 0) { auto r = readln.chomp.split.to!(T[]); r[] += fix; return r; }
T[] _RDA(T = long)(File f, T fix = 0) { auto r = f.readln.chomp.split.to!(T[]); r[] += fix; return r; }
T RD(T = long)() { if (_f.isOpen) return _RD!T(_f); else return _RD!T; }
T[] RDA(T = long)(T fix = 0) { if (_f.isOpen)  return _RDA!T(_f, fix); else return _RDA!T(fix); }
size_t[] MAKE_IDX(alias less = "a < b", Range)(Range range) { auto idx = new size_t[](range.length); makeIndex!(less)(range, idx); return idx;}
size_t MIN_POS(alias less = "a < b", Range)(Range range) { auto r = minPos!(less)(range); return range.length - r.length; }
void chmin(T)(ref T x, T y) { x = min(x, y); } void chmax(T)(ref T x, T y) { x = max(x, y); }
bool inside(T)(T x, T b, T e) { return x >= b && x < e; }
T lcm(T)(T x, T y) { return x * (y / gcd(x, y)); }
long mod = 10^^9 + 7;
//long mod = 998_244_353;
//long mod = 1_000_003;
void moda(ref long x, long y) { x = (x + y) % mod; }
void mods(ref long x, long y) { x = ((x + mod) - (y % mod)) % mod; }
void modm(ref long x, long y) { x = (x * y) % mod; }
void modpow(ref long x, long y) { if (!y) { x = 1; return; } auto t = x; x.modpow(y>>1); x.modm(x); if (y&1) x.modm(t); }
void modd(ref long x, long y) { y.modpow(mod - 2); x.modm(y); }
void main()
{
    auto N = RD;
    auto K = RD;
    auto edges = new long[][][](N);
    foreach (i; 0..N-1)
    {
        auto a = RD-1;
        auto b = RD-1;
        auto c = RD;
        edges[a] ~= [b, c];
        edges[b] ~= [a, c];
    }
    debug writeln(edges);
    auto w = new long[](N-1);
    auto v = new long[](N-1);
    long ans;
    long dfs(long pos, long last, long c)
    {
        if (pos != 0 && edges[pos].length == 1)
        {
            w ~= c;
            v ~= c;
            ans += c;
            return 1;
        }
        long cnt;
        foreach (e; edges[pos])
        {
            if (e[0] == last) continue;
            cnt += dfs(e[0], pos, e[1]);
        }
        if (c != -1)
        {
            w ~= c;
            v ~= c * cnt;
            ans += c * cnt;
        }
        return cnt;
    }
    dfs(0, -1, -1);
    debug writeln(w);
    auto dp = new long[](K+1);
    foreach (i; 0..w.length)
    {
        foreach (j; w[i]..K+1)
        {
            dp[j-w[i]].chmax(dp[j]+v[i]);
        }
    }
    ans += dp.maxElement;
    writeln(ans);
    stdout.flush;
    debug readln;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0