diff --git "a/\353\260\261\354\244\200 14497\353\262\210 \354\243\274\353\202\234\354\235\230 \353\202\234(\351\233\243)/main.py" "b/\353\260\261\354\244\200 14497\353\262\210 \354\243\274\353\202\234\354\235\230 \353\202\234(\351\233\243)/main.py" index 3b45c49..8b0fc5d 100644 --- "a/\353\260\261\354\244\200 14497\353\262\210 \354\243\274\353\202\234\354\235\230 \353\202\234(\351\233\243)/main.py" +++ "b/\353\260\261\354\244\200 14497\353\262\210 \354\243\274\353\202\234\354\235\230 \353\202\234(\351\233\243)/main.py" @@ -7,23 +7,14 @@ def solve(): N, M = map(int, input().strip().split()) junan_y, junan_x, thief_y, thief_x = map(int, input().strip().split()) junan_y, junan_x, thief_y, thief_x = junan_y - 1, junan_x - 1, thief_y - 1, thief_x - 1 - maps = [] - - - for _ in range(N): - maps.append(list(input().strip())) - - queue = deque([[junan_y, junan_x]]) - - dys = [-1, 0, 1, 0] - dxs = [0, 1, 0, -1] - + maps = [list(input().strip()) for _ in range(N)] answer = 1 students = deque() while True: visited = [[False for _ in range(M)] for _ in range(N)] + queue = deque([[junan_y, junan_x]]) while queue: sy, sx = queue.popleft() if visited[sy][sx]: @@ -31,8 +22,8 @@ def solve(): visited[sy][sx] = True - for i in range(len(dys)): - dy, dx = dys[i], dxs[i] + dyxs = [[-1, 0], [0, 1], [1, 0], [0, -1]] + for dy, dx in dyxs: y, x = sy + dy, sx + dx if 0 <= y < N and 0 <= x < M and visited[y][x] == False: @@ -50,8 +41,6 @@ def solve(): answer += 1 - queue.append([junan_y, junan_x]) - if __name__ == '__main__': solve()