結果
問題 |
No.1817 Reversed Edges
|
ユーザー |
|
提出日時 | 2022-01-21 21:35:23 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
RE
|
実行時間 | - |
コード長 | 800 bytes |
コンパイル時間 | 395 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 57,472 KB |
最終ジャッジ日時 | 2024-11-25 22:58:42 |
合計ジャッジ時間 | 14,447 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 RE * 1 |
ソースコード
from collections import deque N = int(input()) T = [[] for _ in range(N)] edges = [tuple(int(x) - 1 for x in input().split()) for _ in range(N - 1)] for i, (a, b) in enumerate(edges): T[a].append((b, i)) T[b].append((a, i)) d = deque([(0, -1)]) rev_edges = [None] * (N - 1) while d: v, p = d.pop() for x, i in T[v]: if x == p: continue rev_edges[i] = (v > x) d.append((x, v)) ans = [0] * N cur = rev_edges.count(True) def rec(v, p): global cur ans[v] = cur for x, i in T[v]: if x == p: continue cur += -1 if rev_edges[i] else 1 rev_edges[i] = not rev_edges[i] rec(x, v) cur += -1 if rev_edges[i] else 1 rev_edges[i] = not rev_edges[i] rec(0, -1) print(*ans, sep='\n')