結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
ユーザー roaris
提出日時 2021-01-23 20:44:19
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,695 bytes
コンパイル時間 577 ms
コンパイル使用メモリ 82,156 KB
実行使用メモリ 474,336 KB
最終ジャッジ日時 2024-12-31 14:07:15
合計ジャッジ時間 35,055 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 21 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**5+100)
from collections import *

def bfs():
    dist = [-1]*(N+1)
    dist[0] = 0
    q = deque([0])
    
    while q:
        v = q.popleft()
        
        for nv in g[v]:
            if dist[nv]==-1:
                dist[nv] = dist[v]+1
                q.append(nv)
    
    return dist

def rbfs():
    dist = [-1]*(N+1)
    dist[N] = 0
    q = deque([N])
    
    while q:
        v = q.popleft()
        
        for nv in rg[v]:
            if dist[nv]==-1:
                dist[nv] = dist[v]+1
                q.append(nv)
    
    return dist

def dfs(v, pv):
    global cycle
    
    seen[v] = True
    
    for nv in g[v]:
        if finished[nv]:
            continue
        
        if seen[nv] and not finished[nv]:
            cycle = True
            return

        dfs(nv, v);
        
        if cycle:
            return
    
    finished[v] = True

N, M = map(int, input().split())
G = [[] for _ in range(N+1)]
g = [[] for _ in range(N+1)]
rg = [[] for _ in range(N+1)]

for _ in range(M):
    u, v, l, a = map(int, input().split())
    G[u].append((v, l, a))
    g[u].append(v)
    rg[v].append(u)

dist1, dist2 = bfs(), rbfs()
seen = [False]*(N+1)
finished = [False]*(N+1)
cycle = False
dfs(0, -1)

if cycle:
    print('INF')
    exit()

dpl = [0]*(N+1)
dpc = [0]*(N+1)
dpc[0] = 1
MOD = 10**9+7
q = deque([0])

while q:
    nq = deque([])
    
    while q:
        v = q.popleft()
        
        for nv, l, a in G[v]:
            dpl[nv] += dpl[v]+dpc[v]*a*l
            dpl[nv] %= MOD
            dpc[nv] += dpc[v]*a
            dpc[nv] %= MOD
            nq.append(nv)
    
    q = nq

print(dpl[N])
0