結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
ユーザー rlangevin
提出日時 2024-01-08 01:29:32
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,479 bytes
コンパイル時間 259 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 158,124 KB
最終ジャッジ日時 2024-09-27 19:34:17
合計ジャッジ時間 12,869 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

from collections import deque
def topsort(G):
    Q = deque()
    N = len(G)
    count = [0] * N
    for u in range(N):
        for v, _, _ in G[u]:
            count[v] += 1

    for u in range(N):
        if count[u] == 0:
            Q.append(u)
    anslst = []
    while Q:
        u = Q.pop()
        anslst.append(u)
        for v, _, _ in G[u]:
            count[v] -= 1
            if count[v] == 0:
                Q.append(v)
    return anslst

def bfs(G, s):
    Q = deque([s])
    N = len(G)
    dist = [-1] * N
    par = [-1] * N
    dist[s] = 0
    while Q:
        u = Q.popleft()
        for v in G[u]:
            if dist[v] != -1:
                continue
            dist[v] = dist[u] + 1
            par[v] = u
            Q.append(v)
    return dist

N, M = map(int, input().split())
G = [[] for i in range(N + 1)]
Gr = [[] for i in range(N + 1)]
D = [] 
for i in range(M):
    u, v, l, a = list(map(int, input().split()))
    D.append([u, v, l, a])
    Gr[v].append(u)
    
dist = bfs(Gr, N)
for u, v, l, a in D:
    if dist[u] == -1 or dist[v] == -1:
        continue
    G[u].append((v, l, a))

dp = [0] * (N + 1)
dp_sum = [0] * (N + 1)
dp[0] = 1
L = topsort(G)
if len(L) != N + 1:
    print("INF")
    exit()

mod = 10**9 + 7    
for u in L:
    for v, l, a in G[u]:
        dp[v] += dp[u] * a
        dp[v] %= mod
        dp_sum[v] += (dp_sum[u] + dp[u] * l) * a
        dp_sum[v] %= mod
        
print(dp_sum[-1])
0