結果
| 問題 | No.3635 Probability trip |
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2026-08-21 22:33:32 |
| 言語 | PyPy3 (7.3.17) |
| 結果 |
AC
|
| 実行時間 | 348 ms / 2,000 ms |
| + 894µs | |
| コード長 | 1,815 bytes |
| 記録 | |
| コンパイル時間 | 225 ms |
| コンパイル使用メモリ | 96,356 KB |
| 実行使用メモリ | 85,376 KB |
| 最終ジャッジ日時 | 2026-08-21 22:34:14 |
| 合計ジャッジ時間 | 9,833 ms |
|
ジャッジサーバーID (参考情報) |
judge3_0 / judge2_0 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 43 |
ソースコード
import sys
# input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################
mod = 998244353
def mat_add(A, B):
assert len(A)==len(B) and len(A[0]) == len(B[0])
n = len(A)
m = len(A[0])
for i in range(n):
for j in range(m):
A[i][j] += B[i][j]
A[i][j] %= mod
return A
def mat_mul(A,B):
assert len(A[0]) == len(B)
n = len(A)
m = len(B[0])
p = len(A[0])
R = [[0 for j in range(m)]for i in range(n)]
for i in range(n):
for j in range(m):
for k in range(p):
R[i][j] += A[i][k]*B[k][j] % mod
R[i][j] %= mod
return R
def mat_pow(A, x):
assert len(A)==len(A[0])
n = len(A)
R = [[0 for j in range(n)]for i in range(n)]
for i in range(n):
R[i][i] = 1
while x > 0:
if x&1:
R = mat_mul(R, A)
A = mat_mul(A,A)
x >>= 1
return R
def mat_pri(A):
for i in A:
print(*i)
def solve(n, g, s, t, k):
res = mat_pow(g, k)
return res[s][t]
n, m = na()
g = [[0] * n for i in range(n)]
deg = [0] * n
for i in range(m):
a, b = na()
a -= 1
b -= 1
g[a][b] += 1
g[b][a] += 1
deg[a] += 1
deg[b] += 1
mod = 998244353
for i in range(n):
if deg[i] == 0:
continue
x = pow(deg[i], mod-2, mod)
for j in range(n):
g[i][j] *= x
g[i][j] %= mod
s, t, a, b = na()
a -= 1
b -= 1
print(solve(n, g, 0, b, t - 1) * solve(n, g, b, a, s - t) * pow(solve(n, g, 0, a, s - 1), mod-2, mod) % mod)