結果

問題 No.277 根掘り葉掘り
ユーザー ckawatakckawatak
提出日時 2017-08-24 23:06:02
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,288 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 59,560 KB
最終ジャッジ日時 2024-04-23 14:13:57
合計ジャッジ時間 5,582 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 30 ms
16,256 KB
testcase_01 AC 31 ms
10,880 KB
testcase_02 WA -
testcase_03 AC 33 ms
10,880 KB
testcase_04 AC 33 ms
10,880 KB
testcase_05 AC 31 ms
10,752 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 33 ms
10,880 KB
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
権限があれば一括ダウンロードができます

ソースコード

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