結果

問題 No.1103 Directed Length Sum
ユーザー rin204
提出日時 2021-07-17 16:10:59
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 586 bytes
コンパイル時間 882 ms
コンパイル使用メモリ 81,996 KB
実行使用メモリ 849,136 KB
最終ジャッジ日時 2024-07-07 03:57:40
合計ジャッジ時間 4,805 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other MLE * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10 ** 9)
MOD = 10 ** 9 + 7

n = int(input())
root = set(range(n))
edges = [[] for _ in range(n)]
for _ in range(n - 1):
    a, b = map(int, input().split())
    edges[a - 1].append(b - 1)
    root.remove(b - 1)

for s in root:
    break

dp = [0] * n
cnt = [0] * n

def dfs(pos, bpos):
    cnt[pos] = 1
    for npos in edges[pos]:
        if npos == bpos:
            continue
        dfs(npos, pos)
        dp[pos] += dp[npos] + cnt[npos]
        cnt[pos] += cnt[npos]
        dp[pos] %= MOD
        cnt[pos] %= MOD
    
dfs(s, -1)
print(sum(dp) % MOD)
0