結果

問題 No.1364 [Renaming] Road to Cherry from Zelkova
コンテスト
ユーザー norioc
提出日時 2026-10-04 22:26:27
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 618 ms / 2,500 ms
+ 574µs
コード長 1,591 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,006 ms
コンパイル使用メモリ 82,604 KB
実行使用メモリ 291,164 KB
最終ジャッジ日時 2026-10-04 22:26:51
合計ジャッジ時間 21,140 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge4_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import defaultdict, deque
from atcoder.scc import SCCGraph
from itertools import dropwhile


def scc(n: int, adj: dict[int, list[int]]) -> list[list[int]]:
    """グラフを強連結成分分解する。

    Args:
        n: 頂点数
        adj: 隣接リスト {u: [v...]}

    Returns:
        強連結成分のリスト (トポロジカル順)
    """

    g = SCCGraph(n)
    for u, vs in adj.items():
        for v in vs:
            g.add_edge(u, v)

    return g.scc()


MOD = 10**9 + 7
N, M = map(int, input().split())

adj = defaultdict(list)
adj2 = defaultdict(list)
r_adj = defaultdict(list)  # 逆辺
for _ in range(M):
    u, v, l, a = map(int, input().split())
    adj[u].append(v)
    adj2[u].append((v, l, a))

    r_adj[v].append(u)


def calc_reachables():
    reachables = [False] * (N+1)  # 頂点 N へ到達可能な頂点
    reachables[N] = True

    q = deque([N])
    while q:
        v = q.popleft()
        for to in r_adj[v]:
            if reachables[to]: continue
            reachables[to] = True
            q.append(to)

    return reachables


reachables = calc_reachables()

tots = [0] * (N+1)
cnts = [0] * (N+1)
cnts[0] = 1

cc = scc(N+1, adj)
for g in dropwhile(lambda g: 0 not in g, cc):
    if len(g) > 1:
        for v in g:
            if reachables[v]:
                print('INF')
                exit()

    for v in g:
        for to, l, a in adj2[v]:
            tots[to] += (tots[v] + cnts[v] * l) * a
            tots[to] %= MOD
            cnts[to] += cnts[v] * a
            cnts[to] %= MOD

ans = tots[N]
print(ans)
0