結果

問題 No.3504 Insert Maze
コンテスト
ユーザー ,
提出日時 2026-04-18 23:20:34
言語 Python3
(3.14.3 + numpy 2.4.4 + scipy 1.17.1)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
WA  
実行時間 -
コード長 3,520 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 514 ms
コンパイル使用メモリ 20,700 KB
実行使用メモリ 177,908 KB
最終ジャッジ日時 2026-04-18 23:23:35
合計ジャッジ時間 12,148 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 WA * 11 TLE * 1 -- * 67
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
import heapq

def solve():
    data = sys.stdin.read().split()
    height = int(data[0])
    width = int(data[1])
    grid = [data[i + 2] for i in range(height)]

    INF = float('inf')
    distances = {}
    priority_queue = []

    def update_state(state, cost):
        if distances.get(state, INF) > cost:
            distances[state] = cost
            heapq.heappush(priority_queue, (cost, state))

    update_state(('R', 0, 0, True, True), 0)

    while priority_queue:
        current_cost, state = heapq.heappop(priority_queue)

        if current_cost > distances.get(state, INF):
            continue

        state_type = state[0]

        if state_type == 'R':
            _, row, col, can_row, can_col = state

            if row == height - 1 and col == width - 1:
                print(current_cost)
                return

            for dr, dc in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
                new_r, new_c = row + dr, col + dc
                if 0 <= new_r < height and 0 <= new_c < width:
                    if grid[new_r][new_c] != '#':
                        update_state(('R', new_r, new_c, can_row, can_col), current_cost + 1)

            if can_row:
                if row > 0:
                    update_state(('VR', row - 1, col, can_col), current_cost + 1)
                if row < height - 1:
                    update_state(('VR', row, col, can_col), current_cost + 1)

            if can_col:
                if col > 0:
                    update_state(('VC', row, col - 1, can_row), current_cost + 1)
                if col < width - 1:
                    update_state(('VC', row, col, can_row), current_cost + 1)

        elif state_type == 'VR':
            _, row, col, can_col = state

            for dc in [-1, 1]:
                new_c = col + dc
                if 0 <= new_c < width:
                    update_state(('VR', row, new_c, can_col), current_cost + 1)

            if grid[row][col] != '#':
                update_state(('R', row, col, False, can_col), current_cost + 1)
            if row + 1 < height and grid[row + 1][col] != '#':
                update_state(('R', row + 1, col, False, can_col), current_cost + 1)

            if can_col:
                update_state(('VX', row, col), current_cost + 1)

        elif state_type == 'VC':
            _, row, col, can_row = state

            for dr in [-1, 1]:
                new_r = row + dr
                if 0 <= new_r < height:
                    update_state(('VC', new_r, col, can_row), current_cost + 1)

            if grid[row][col] != '#':
                update_state(('R', row, col, can_row, False), current_cost + 1)
            if col + 1 < width and grid[row][col + 1] != '#':
                update_state(('R', row, col + 1, can_row, False), current_cost + 1)

            if can_row:
                update_state(('VX', row, col), current_cost + 1)

        else:
            _, row, col = state

            for dr, dc in [(-1,0),(1,0),(0,-1),(0,1)]:
                new_r, new_c = row + dr, col + dc
                if 0 <= new_r < height and 0 <= new_c < width:
                    update_state(('VX', new_r, new_c), current_cost + 1)

            for r2 in [row, row + 1]:
                for c2 in [col, col + 1]:
                    if 0 <= r2 < height and 0 <= c2 < width:
                        if grid[r2][c2] != '#':
                            update_state(('R', r2, c2, False, False), current_cost + 1)

    print(-1)

if __name__ == "__main__":
    solve()
0