結果

問題 No.659 徘徊迷路
ユーザー roarisroaris
提出日時 2019-11-02 20:27:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,341 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 85,220 KB
最終ジャッジ日時 2024-09-14 23:16:17
合計ジャッジ時間 21,273 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 131 ms
76,752 KB
testcase_01 AC 623 ms
78,720 KB
testcase_02 AC 94 ms
76,544 KB
testcase_03 AC 63 ms
69,504 KB
testcase_04 AC 916 ms
80,352 KB
testcase_05 AC 165 ms
76,952 KB
testcase_06 AC 161 ms
77,096 KB
testcase_07 AC 102 ms
76,544 KB
testcase_08 AC 929 ms
80,060 KB
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 AC 544 ms
78,336 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