結果
問題 | No.1364 [Renaming] Road to Cherry from Zelkova |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
import sysinput = sys.stdin.readlinefrom collections import dequedef topsort(G):Q = deque()N = len(G)count = [0] * Nfor u in range(N):for v, _, _ in G[u]:count[v] += 1for 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] -= 1if count[v] == 0:Q.append(v)return anslstdef bfs(G, s):Q = deque([s])N = len(G)dist = [-1] * Npar = [-1] * Ndist[s] = 0while Q:u = Q.popleft()for v in G[u]:if dist[v] != -1:continuedist[v] = dist[u] + 1par[v] = uQ.append(v)return distN, 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:continueG[u].append((v, l, a))dp = [0] * (N + 1)dp_sum = [0] * (N + 1)dp[0] = 1L = topsort(G)if len(L) != N + 1:print("INF")exit()mod = 10**9 + 7for u in L:for v, l, a in G[u]:dp[v] += dp[u] * adp[v] %= moddp_sum[v] += (dp_sum[u] + dp[u] * l) * adp_sum[v] %= modprint(dp_sum[-1])