結果

問題 No.277 根掘り葉掘り
ユーザー ckawatak
提出日時 2017-08-24 23:06:02
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,288 bytes
コンパイル時間 105 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 60,876 KB
最終ジャッジ日時 2024-10-15 14:21:57
合計ジャッジ時間 6,273 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 4 WA * 3 TLE * 1 -- * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Node:
    def __init__(self, id):
        self.id = id
        self.children = []
        self.parent = 0
        self.r = 0
        self.l = 0

N = int(input())

nodes = [None] * (N+1)

for n in range(N-1):
    x,y = list(map(int, input().split(' ')))
    if nodes[x] == None:
        X = Node(x)
        nodes[x] = X        
    nodes[x].children.append(y)
    if nodes[y] == None:
        Y = Node(y)
        nodes[y] = Y
    nodes[y].parent = x
nodes[0] = Node(0)

# find distance from the root
parents = {}
for n in range(1,N+1):
    node = nodes[n]
    if node != None:
        r = 0
        parents[n] = []
        while node.parent != 0:
            parents[n].append(node.parent)
            node = nodes[node.parent]
            r += 1
        nodes[n].r = r

# find leaves
leaves = []    
for n in range(1,N+1):
    node = nodes[n]
    if node != None and len(node.children) == 0:
        leaves.append(n)
leaves.sort()

# find distance from the leaf
for n in range(1,N+1):
    node = nodes[n]
    if node != None:
        for l in range(len(leaves)):
            if n in parents[leaves[l]]:
                node.l = parents[leaves[l]].index(n) + 1
                break

for n in range(1,N+1):
    node = nodes[n]
    if node != None:
        print(min(node.r, node.l))
0