結果

問題 No.659 徘徊迷路
ユーザー roarisroaris
提出日時 2019-11-02 20:27:30
言語 PyPy3
(7.0.0)
結果
TLE   .
実行時間 -
コード長 1,341 Byte
コンパイル時間 250 ms
使用メモリ 73,232 KB
最終ジャッジ日時 2020-07-02 12:28:46

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 203 ms
69,784 KB
testcase_01 AC 704 ms
72,244 KB
testcase_02 AC 164 ms
69,356 KB
testcase_03 AC 133 ms
68,296 KB
testcase_04 AC 1,000 ms
73,232 KB
testcase_05 AC 239 ms
69,596 KB
testcase_06 AC 229 ms
70,180 KB
testcase_07 AC 172 ms
69,052 KB
testcase_08 AC 1,050 ms
72,976 KB
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 AC 638 ms
71,616 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