結果

問題 No.1817 Reversed Edges
ユーザー marroncastlemarroncastle
提出日時 2022-01-21 23:23:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 339 ms / 2,000 ms
コード長 1,307 bytes
コンパイル時間 337 ms
コンパイル使用メモリ 87,056 KB
実行使用メモリ 103,116 KB
最終ジャッジ日時 2023-08-17 08:07:35
合計ジャッジ時間 8,103 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 91 ms
71,524 KB
testcase_01 AC 95 ms
71,664 KB
testcase_02 AC 92 ms
71,660 KB
testcase_03 AC 98 ms
71,832 KB
testcase_04 AC 92 ms
71,712 KB
testcase_05 AC 91 ms
71,584 KB
testcase_06 AC 95 ms
71,596 KB
testcase_07 AC 296 ms
97,124 KB
testcase_08 AC 192 ms
88,900 KB
testcase_09 AC 278 ms
95,932 KB
testcase_10 AC 206 ms
88,032 KB
testcase_11 AC 234 ms
92,660 KB
testcase_12 AC 311 ms
99,288 KB
testcase_13 AC 332 ms
101,888 KB
testcase_14 AC 333 ms
103,116 KB
testcase_15 AC 339 ms
99,236 KB
testcase_16 AC 311 ms
98,844 KB
testcase_17 AC 317 ms
101,368 KB
testcase_18 AC 331 ms
102,324 KB
testcase_19 AC 334 ms
101,024 KB
testcase_20 AC 337 ms
101,968 KB
testcase_21 AC 318 ms
99,692 KB
testcase_22 AC 189 ms
100,116 KB
testcase_23 AC 194 ms
101,204 KB
testcase_24 AC 187 ms
98,992 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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] + int(u>v)*2-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] + int(u<v)

N = int(input())
G = Tree(N)
G.add_edges(ind=1, bi=True)
G.dp(0)
print(*[G.dp_back[0]+G.dp_for[i]for i in range(N)],sep='\n')
0