結果

問題 No.1103 Directed Length Sum
ユーザー marroncastlemarroncastle
提出日時 2020-07-03 22:31:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 538 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-09-17 04:10:15
合計ジャッジ時間 2,571 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 2
other RE * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
sys.setrecursionlimit(10**10)

def dfs(v,nums):
    nums[v] = [0,0]
    if len(edge[v])==0:
        return
    for u in edge[v]:
        if nums[u] == -1:
            dfs(u,nums)
        nums[v][0] += sum(nums[u])+1
        nums[v][1] += nums[u][1]+1
    return

N = int(input())
edge = [[] for _ in range(N)]
for i in range(N-1):
    a,b = map(int, input().split())
    edge[a-1].append(b-1)
nums = [-1]*N
for i in range(N):
    if nums[i] == -1:
        dfs(i,nums)
ans = 0
for i in range(N):
    ans += nums[i][0]
print(ans)
0