結果

問題 No.1221 木 *= 3
ユーザー TANIGUCHI KousukeTANIGUCHI Kousuke
提出日時 2020-09-09 14:18:49
言語 Ruby
(3.3.0)
結果
AC  
実行時間 622 ms / 2,000 ms
コード長 896 bytes
コンパイル時間 463 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 35,276 KB
最終ジャッジ日時 2024-05-08 21:38:16
合計ジャッジ時間 12,341 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 90 ms
12,160 KB
testcase_01 AC 91 ms
12,160 KB
testcase_02 AC 90 ms
12,032 KB
testcase_03 AC 90 ms
12,032 KB
testcase_04 AC 92 ms
12,160 KB
testcase_05 AC 91 ms
12,160 KB
testcase_06 AC 91 ms
12,032 KB
testcase_07 AC 583 ms
33,272 KB
testcase_08 AC 578 ms
32,896 KB
testcase_09 AC 590 ms
33,128 KB
testcase_10 AC 588 ms
33,268 KB
testcase_11 AC 594 ms
33,100 KB
testcase_12 AC 572 ms
34,244 KB
testcase_13 AC 577 ms
33,860 KB
testcase_14 AC 580 ms
34,040 KB
testcase_15 AC 573 ms
34,172 KB
testcase_16 AC 575 ms
34,032 KB
testcase_17 AC 608 ms
35,276 KB
testcase_18 AC 619 ms
34,932 KB
testcase_19 AC 622 ms
35,148 KB
testcase_20 AC 621 ms
35,192 KB
testcase_21 AC 619 ms
35,140 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N = gets.to_i
A = gets.split.map(&:to_i)
B = gets.split.map(&:to_i)
G = Array.new(N + 1){ [] }

def max(a,b); a > b ? a : b; end
class << G
  def bfs_order
    stack = [1]
    used = Array.new(size)
    order = []
    until stack.empty?
      u = stack.pop
      
      next if used[u]
      
      used[u] = order.size
      order << u
      self[u].each do |v|
        next if used[v]
        stack << v
      end
    end
    order.reverse
  end
end

(1 ... N).each do 
  a,b = gets.split.map(&:to_i)
  G[a] << b
  G[b] << a
end

order = G.bfs_order

dp0 = Array.new(N + 1)
dp1 = Array.new(N + 1)
used = Array.new(N + 1)

order.each do |u|
  used[u] = true
  
  dp0[u] = G[u].map do |v|
    next 0 if !used[v]
    max(dp0[v], dp1[v])
  end.sum + A[u - 1]
  
  dp1[u] = G[u].map do |v|
    next 0 if !used[v]
    max(dp1[v] + B[v - 1] + B[u - 1], dp0[v])
  end.sum
end

puts max(dp0[1], dp1[1])
0