結果
問題 | No.1817 Reversed Edges |
ユーザー | marroncastle |
提出日時 | 2022-01-21 23:01:49 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,439 bytes |
コンパイル時間 | 233 ms |
コンパイル使用メモリ | 82,664 KB |
実行使用メモリ | 103,040 KB |
最終ジャッジ日時 | 2024-05-04 14:22:55 |
合計ジャッジ時間 | 4,980 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 39 ms
53,760 KB |
testcase_01 | AC | 39 ms
53,760 KB |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | AC | 37 ms
54,016 KB |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | AC | 132 ms
102,400 KB |
testcase_23 | AC | 134 ms
103,040 KB |
testcase_24 | AC | 126 ms
98,944 KB |
ソースコード
import sys input = sys.stdin.readline from collections import deque sys.setrecursionlimit(10**6) class Tree: def __init__(self, N): self.V = N self.edge = [[] for _ in range(N)] self.order = [] def add_edges(self, ind=1, bi=True): for i in range(self.V-1): a,b = map(int, input().split()) a -= ind; b -= ind self.edge[a].append(b) if bi: self.edge[b].append(a) def dp(self, start): stack = deque([start]) self.parent = [self.V]*self.V; self.parent[start] = -1 self.order.append(start) #記録したい値の配列を定義 self.dp_for = [0]*self.V self.dp_back = [0]*self.V while stack: v = stack.pop() for u in self.edge[v]: if u==self.parent[v]: continue self.parent[u]=v stack.append(u); self.order.append(u) self.dp_for[u] += self.dp_for[v] if u > v: self.dp_for[u] += 1 for v in self.order[::-1]: for u in self.edge[v]: if u==self.parent[v]: continue self.dp_back[v] += self.dp_back[u] #帰りがけ処理 if u < v: self.dp_back[v] += 1 N = int(input()) G = Tree(N) G.add_edges(ind=1, bi=True) ans=[0]*N G.dp(0) for i in range(N): ans[i]=G.dp_back[0]+G.dp_for[i] print(*ans, sep='\n')