結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
コンテスト
ユーザー tktk_snsn
提出日時 2021-01-22 21:57:43
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 973 bytes
コンパイル時間 403 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 884,332 KB
最終ジャッジ日時 2024-12-28 00:17:34
合計ジャッジ時間 41,627 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 MLE * 1
other AC * 31 WA * 2 TLE * 7 MLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
mod = 10**9+7

N, M = map(int, input().split())
deg = [0] * (N+1)
go = [[] for _ in range(N+1)]
back = [[] for _ in range(N+1)]
for _ in range(M):
    a, b, c, d = map(int, input().split())
    go[a].append((b, c, d))
    back[b].append((a, c, d))
    deg[b] += 1

topo = []
q = []
for i, d in enumerate(deg):
    if d == 0:
        q.append(i)
while q:
    s = q.pop()
    topo.append(s)
    for t, _, _ in go[s]:
        deg[t] -= 1
        if deg[t] == 0:
            q.append(t)

if len(topo) < N + 1:
    print("INF")
    exit()


dp = [defaultdict(int) for _ in range(N + 1)]
dp[N][0] += 1  # len, cnt
for s in topo[::-1][:-1]:
    for k, v in dp[s].items():
        for t, dist, num in back[s]:
            dp[t][k + dist] += v * num % mod
            dp[t][k + dist] %= mod

ans = 0
for k, v in dp[0].items():
    ans += k * v
    ans %= mod

print(ans)
0