結果
問題 |
No.424 立体迷路
|
ユーザー |
|
提出日時 | 2017-01-19 08:54:05 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,336 bytes |
コンパイル時間 | 124 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 11,776 KB |
最終ジャッジ日時 | 2024-12-23 01:29:14 |
合計ジャッジ時間 | 2,123 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 19 WA * 2 |
ソースコード
# 入力 h, w = map(int, input().split()) sx, sy, gx, gy = map(lambda x: x - 1, map(int, input().split())) # sx, sy, gx, gy = sx - 1, sy - 1, gx - 1, gy - 1 b = [[] for i in range(h)] for i in range(h): line = input() for takasa in line: b[i].append(int(takasa)) # print(b) dx = [1, 0, -1, 0, 2, 0, -2, 0,] dy = [0, 1, 0 ,-1, 0, 2, 0, -2,] ikeru = [[None] * w for i in range(h)] # print(ikeru) ikeru[sx][sy] = True tugis = [(sx, sy)] tansaku_zumi = set() # print(tugis[0][0]) while tugis: cur = tugis.pop() for i in range(len(dx)): if 0 <= cur[0] + dx[i] < h and 0 <= cur[1] + dy[i] < w: if i > 3 and b[cur[0]][cur[1]] == b[cur[0] + dx[i]][cur[1] + dy[i]]: if (cur[0] + dx[i], cur[1] + dy[i]) not in tansaku_zumi: ikeru[cur[0] + dx[i]][cur[1] + dy[i]] = True tugis.append((cur[0] + dx[i], cur[1] + dy[i])) elif i <= 3 and abs(b[cur[0] + dx[i]][cur[1] + dy[i]] - b[cur[0]][cur[1]]) <= 1: if (cur[0] + dx[i], cur[1] + dy[i]) not in tansaku_zumi: ikeru[cur[0] + dx[i]][cur[1] + dy[i]] = True tugis.append((cur[0] + dx[i], cur[1] + dy[i])) tansaku_zumi.add(cur) # print(tansaku_zumi) if (gx, gy) in tansaku_zumi: print('YES') else: print('NO')