結果
問題 | No.659 徘徊迷路 |
ユーザー |
![]() |
提出日時 | 2021-03-12 09:12:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 163 ms / 2,000 ms |
コード長 | 1,163 bytes |
コンパイル時間 | 227 ms |
コンパイル使用メモリ | 82,244 KB |
実行使用メモリ | 77,348 KB |
最終ジャッジ日時 | 2024-10-13 18:28:35 |
合計ジャッジ時間 | 2,607 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 12 |
ソースコード
def matmul(A,B): # A,B: 行列 res = [[0.0]*len(B[0]) for _ in range(len(A))] for i, resi in enumerate(res): for k, aik in enumerate(A[i]): for j,bkj in enumerate(B[k]): resi[j] += aik*bkj return res def matpow(A,p): #A^p mod M if p%2: return matmul(A, matpow(A,p-1)) elif p > 0: b = matpow(A,p//2) return matmul(b,b) else: return [[int(i==j)*1.0 for j in range(len(A))] for i in range(len(A))] h,w,t = map(int,input().split()) sx,sy = map(int,input().split()) gx,gy = map(int,input().split()) b = [input() for _ in range(h)] N = h*w A = [[0.0]*N for _ in range(N)] for i in range(1,h-1): for j in range(1,w-1): if b[i][j] == "#": continue v = i*w+j c = 0 for ni,nj in [(i-1,j),(i+1,j),(i,j-1),(i,j+1)]: if b[ni][nj] == ".": c += 1 if c==0: A[v][v] = 1.0 continue for ni,nj in [(i-1,j),(i+1,j),(i,j-1),(i,j+1)]: if b[ni][nj] == ".": A[v][ni*w+nj] = 1/c B = matpow(A,t) v = [0.0]*N v[sx*w+sy] = 1.0 v = matmul([v],B) print(v[0][gx*w+gy])