結果

問題 No.1488 Max Score of the Tree
ユーザー simansiman
提出日時 2021-04-27 14:18:56
言語 Ruby
(3.3.0)
結果
AC  
実行時間 912 ms / 2,000 ms
コード長 756 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 95,180 KB
最終ジャッジ日時 2024-07-06 08:34:52
合計ジャッジ時間 10,407 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 336 ms
61,312 KB
testcase_01 AC 912 ms
89,988 KB
testcase_02 AC 709 ms
75,008 KB
testcase_03 AC 613 ms
72,448 KB
testcase_04 AC 726 ms
65,664 KB
testcase_05 AC 94 ms
12,544 KB
testcase_06 AC 98 ms
13,312 KB
testcase_07 AC 258 ms
38,144 KB
testcase_08 AC 97 ms
12,800 KB
testcase_09 AC 346 ms
53,120 KB
testcase_10 AC 132 ms
21,376 KB
testcase_11 AC 813 ms
95,180 KB
testcase_12 AC 92 ms
12,544 KB
testcase_13 AC 233 ms
28,544 KB
testcase_14 AC 196 ms
28,416 KB
testcase_15 AC 126 ms
20,096 KB
testcase_16 AC 93 ms
12,416 KB
testcase_17 AC 110 ms
15,744 KB
testcase_18 AC 318 ms
50,048 KB
testcase_19 AC 119 ms
17,664 KB
testcase_20 AC 112 ms
16,256 KB
testcase_21 AC 94 ms
12,416 KB
testcase_22 AC 143 ms
23,168 KB
testcase_23 AC 93 ms
12,672 KB
testcase_24 AC 94 ms
12,416 KB
testcase_25 AC 91 ms
12,416 KB
testcase_26 AC 93 ms
12,544 KB
testcase_27 AC 97 ms
13,056 KB
testcase_28 AC 106 ms
14,720 KB
testcase_29 AC 126 ms
20,096 KB
testcase_30 AC 161 ms
28,032 KB
testcase_31 AC 633 ms
72,704 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N, K = gets.split.map(&:to_i)
E = Hash.new { |h, k| h[k] = [] }
C = Array.new(N + 1) { Array.new(N + 1, 0) }
L = []

(N - 1).times do
  a, b, c = gets.split.map(&:to_i)

  if a < b
    L << [a, b, c]
    E[a] << [b, c]
    E[b] << [a, c]
  else
    L << [b, a, c]
    E[a] << [b, c]
    E[b] << [a, c]
  end
end

def dfs(v, parent = -1)
  val = 0

  E[v].each do |u, c|
    next if u == parent

    a = dfs(u, v)
    C[v][u] = C[u][v] = a
    val += a
  end

  [1, val].max
end

dfs(1)

dp = Hash.new(0)
dp[0] = L.map { |u, v, c| C[u][v] * c }.sum

L.each do |u, v, c|
  temp = dp.dup

  dp.each do |k, val|
    nk = k + c
    nc = val + C[u][v] * c
    next if nk > K
    next if temp[nk] >= nc

    temp[nk] = nc
  end

  dp = temp
end

pp dp.values.max
0