結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
コンテスト
ユーザー tktk_snsn
提出日時 2021-01-22 22:47:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,622 bytes
コンパイル時間 254 ms
コンパイル使用メモリ 82,320 KB
実行使用メモリ 141,436 KB
最終ジャッジ日時 2024-12-28 04:27:29
合計ジャッジ時間 15,254 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 WA * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

N, M = map(int, input().split())
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))


# 0から到達可能なノードを調べる
# 0-Nが非連結なら0
# 到達可能内でサイクルがあればINF
vis0 = [0] * (N + 1)
vis0[0] = 1
q = [0]
while q:
    s = q.pop()
    for t, _, _ in go[s]:
        if not vis0[t]:
            vis0[t] = 1
            q.append(t)

visN = [0] * (N + 1)
visN[N] = 1
q = [N]
while q:
    s = q.pop()
    for t, _, _ in back[s]:
        if not visN[t]:
            visN[t] = 1
            q.append(t)

ok = set(i for i in range(N + 1) if vis0[i] and visN[i])
if not N in ok:
    print(0)
    exit()

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

if len(topo) < len(ok):
    print("INF")
    exit()

cnt = [0] * (N + 1)
cnt[N] = 1
for s in topo[::-1]:
    for t, _, num in back[s]:
        if t in ok:
            cnt[t] += cnt[s] * num
            cnt[t] %= mod

ans = 0
for s in topo:
    for t, L, num in go[s]:
        if t in ok:
            ans += L * num * cnt[t]
            ans %= mod

print(ans)
0