結果

問題 No.659 徘徊迷路
ユーザー roarisroaris
提出日時 2019-11-02 20:27:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,341 bytes
コンパイル時間 1,050 ms
コンパイル使用メモリ 86,636 KB
実行使用メモリ 86,988 KB
最終ジャッジ日時 2023-10-13 01:28:25
合計ジャッジ時間 21,621 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 153 ms
78,492 KB
testcase_01 AC 640 ms
81,492 KB
testcase_02 AC 117 ms
78,056 KB
testcase_03 AC 90 ms
76,716 KB
testcase_04 AC 930 ms
81,756 KB
testcase_05 AC 184 ms
79,564 KB
testcase_06 AC 179 ms
78,860 KB
testcase_07 AC 122 ms
77,848 KB
testcase_08 AC 956 ms
82,004 KB
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 AC 568 ms
80,340 KB
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

def mat_mul(A, B):
    C = [[0]*len(B[0]) for _ in range(len(A))]
    
    for i in range(len(A)):
        for j in range(len(B[0])):
            C[i][j] = sum(A[i][k]*B[k][j] for k in range(len(A[0])))

    return C

def mat_pow(A, n):
    res = [[0]*len(A) for _ in range(len(A))]
    
    for i in range(len(A)):
        res[i][i] = 1
    
    while n > 0:
        if n&1:
            res = mat_mul(A, res)
        
        n >>= 1
        A = mat_mul(A, A)
    
    return res

R, C, T = map(int, input().split())
Sx, Sy = map(int, input().split())
Gx, Gy = map(int, input().split())
B = [input() for _ in range(R)]
#変換行列Tr: Tr[i][j]: jからiへ移動する確率
Tr = [[0]*(R*C) for _ in range(R*C)]

for i in range(R):
    for j in range(C):
        if B[i][j] == '#':
            continue
        
        cnt = 0
        
        for ni, nj in [(i-1, j), (i+1, j), (i, j-1), (i, j+1)]:
            if B[ni][nj] == '.':
                cnt += 1
        
        if cnt == 0:
            Tr[C*i+j][C*i+j] = 1
            continue
        
        p = 1/cnt
        
        for ni, nj in [(i-1, j), (i+1, j), (i, j-1), (i, j+1)]:
            if B[ni][nj] == '.':
                Tr[C*ni+nj][C*i+j] = p

before = [[0] for _ in range(R*C)]
before[C*Sx+Sy][0] = 1
after = mat_mul(mat_pow(Tr, T), before)

print(after[C*Gx+Gy][0])
0