結果

問題 No.1817 Reversed Edges
ユーザー simansiman
提出日時 2023-02-14 10:35:09
言語 Ruby
(3.3.0)
結果
RE  
実行時間 -
コード長 800 bytes
コンパイル時間 494 ms
コンパイル使用メモリ 11,132 KB
実行使用メモリ 32,304 KB
最終ジャッジ日時 2023-09-23 19:47:12
合計ジャッジ時間 13,509 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 80 ms
15,232 KB
testcase_01 AC 78 ms
15,116 KB
testcase_02 AC 78 ms
15,116 KB
testcase_03 AC 77 ms
15,176 KB
testcase_04 AC 83 ms
15,220 KB
testcase_05 AC 80 ms
15,092 KB
testcase_06 AC 80 ms
15,084 KB
testcase_07 AC 482 ms
30,868 KB
testcase_08 AC 275 ms
22,760 KB
testcase_09 AC 460 ms
30,028 KB
testcase_10 AC 304 ms
23,824 KB
testcase_11 AC 361 ms
27,348 KB
testcase_12 AC 539 ms
31,988 KB
testcase_13 AC 526 ms
32,304 KB
testcase_14 AC 539 ms
32,044 KB
testcase_15 AC 542 ms
31,936 KB
testcase_16 AC 543 ms
32,020 KB
testcase_17 AC 555 ms
32,240 KB
testcase_18 AC 556 ms
32,024 KB
testcase_19 AC 551 ms
31,908 KB
testcase_20 AC 527 ms
32,176 KB
testcase_21 AC 512 ms
31,972 KB
testcase_22 AC 406 ms
30,348 KB
testcase_23 AC 404 ms
30,268 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