結果

問題 No.1817 Reversed Edges
ユーザー simansiman
提出日時 2023-02-14 10:35:09
言語 Ruby
(3.3.0)
結果
RE  
実行時間 -
コード長 800 bytes
コンパイル時間 345 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 30,080 KB
最終ジャッジ日時 2024-07-16 19:35:48
合計ジャッジ時間 12,136 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 84 ms
12,160 KB
testcase_01 AC 84 ms
12,288 KB
testcase_02 AC 82 ms
12,288 KB
testcase_03 AC 82 ms
12,288 KB
testcase_04 AC 87 ms
12,160 KB
testcase_05 AC 83 ms
12,160 KB
testcase_06 AC 84 ms
12,160 KB
testcase_07 AC 470 ms
27,904 KB
testcase_08 AC 266 ms
20,352 KB
testcase_09 AC 462 ms
27,136 KB
testcase_10 AC 293 ms
20,864 KB
testcase_11 AC 370 ms
22,656 KB
testcase_12 AC 553 ms
29,952 KB
testcase_13 AC 559 ms
29,952 KB
testcase_14 AC 553 ms
29,952 KB
testcase_15 AC 554 ms
30,080 KB
testcase_16 AC 578 ms
30,080 KB
testcase_17 AC 562 ms
29,952 KB
testcase_18 AC 560 ms
29,952 KB
testcase_19 AC 563 ms
30,080 KB
testcase_20 AC 566 ms
29,824 KB
testcase_21 AC 553 ms
29,952 KB
testcase_22 AC 442 ms
28,672 KB
testcase_23 AC 451 ms
28,800 KB
testcase_24 RE -
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N = gets.to_i
E = Array.new(N + 1) { [] }

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

  E[a] << b
  E[b] << a
end

DP1 = Array.new(N + 1) { [] }

def dfs(v, parent, ans)
  r_cnt = 0

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

    r = dfs(u, v, ans)
    DP1[v] << r
    r_cnt += r
  end

  ans[v] += r_cnt

  if v < parent
    r_cnt += 1
  end

  r_cnt
end

def r_dfs(v, cnt, parent, ans)
  if parent != -1
    ans[v] += cnt
  end

  idx = 0
  E[v].each_with_index do |u|
    next if u == parent

    if v < u
      cnt += 1
    end

    nv = ans[v] - DP1[v][idx]
    nv += 1 if v < u
    r_dfs(u, nv, v, ans)
    idx += 1

    if v < u
      cnt -= 1
    end
  end
end

ans = Array.new(N + 1, 0)
root = 1
dfs(root, -1, ans)
r_dfs(root, ans[root], -1, ans)

1.upto(N) do |v|
  puts ans[v]
end
0