結果

問題 No.867 避難経路
ユーザー IJKTanabe
提出日時 2019-09-17 21:25:44
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,917 bytes
コンパイル時間 651 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 76,544 KB
最終ジャッジ日時 2024-07-07 13:49:15
合計ジャッジ時間 15,630 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 15 TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

def hinan_keiro(H, W, gy, gx, org_grid, queries):
    H = H + 2
    W = W + 2
    goal = (gy * W) + gx
    result = []
    
    for query in queries:
        sy, sx, k = query
        start = (sy * W) + sx
        k = k**2
        
        # initialize calc grid
        calc_grid = [0]*W
        for row in org_grid:
            calc_grid += [0] + list(map(lambda val: val + k, row)) + [0]
        calc_grid += [0]*W
        
        # initialize count
        count = calc_grid[start] - 1
        calc_grid[start] = 1
        
        # initialize decrease points
        dec_points = tuple([start])
        
        while calc_grid[goal] != 0:
            # debug -----
            # for r in range(0, H):
            #     print(calc_grid[r * W:r * W + W])   
            # input()
            # -----

            # copy decrease points to buffer
            buffer = set(dec_points)

            for point in dec_points:
                calc_grid[point] -= 1
                if calc_grid[point] == 0:
                    buffer.remove(point)
                    for new_point in [point - 1, point + 1, point - W, point + W]:
                        if calc_grid[new_point] > 0:
                            buffer.add(new_point)
            dec_points = tuple(buffer)                 
            count += 1
        result.append(count)
    return result

if __name__ == '__main__':
    H, W = map(int, input().split(' '))
    gy, gx = map(int, input().split(' '))

    grid = tuple([
        tuple(map(int, input().split(' ')))
        for i in range(H)])

    Q = int(input())
    queries = tuple([
        tuple(map(int, input().split(' ')))
        for i in range(Q)])
    
    # import time
    # start_time = time.time()

    result = hinan_keiro(H, W, gy, gx, grid, queries)
    for i in result: print(i)

    # process_time = time.time() - start_time
    # print('process_time = ' + str(process_time))
0