結果

問題 No.1817 Reversed Edges
コンテスト
ユーザー marroncastle
提出日時 2022-01-21 23:01:49
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,439 bytes
コンパイル時間 572 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 102,980 KB
最終ジャッジ日時 2024-11-26 02:23:28
合計ジャッジ時間 6,172 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 4 WA * 19
権限があれば一括ダウンロードができます

ソースコード

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]
                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')
0