6098 : 성실한 개미

문제

0(갈 수 있는 곳), 1(벽 또는 장애물), 2(먹이)로 구성된 미로 상자에 오른쪽 또는 아래로만 움직일 수 있는 개미가

갇힐 때(오른쪽 길이 나타나면 다시 오른쪽으로 이동) 개미의 이동 경로를 9로 표현하는 프로그램

 

개미는 항상 (2,2)에서 출발하고, 더 이상 움직일 수 없거나 먹이를 찾은 경우 이동을 멈춘다.

---

board = []
x = 1
y = 1

for _ in range(10):
    maze = list(map(int, input().split()))
    board.append(maze)
while True:
    if board[x][y] == 2:
        board[x][y] = 9
        break
    elif board[x+1][y] == 1 and board[x][y+1] == 1:
        board[x][y] = 9
        break
    board[x][y] = 9
    if board[x][y+1] == 1:
        x+=1
    elif board[x+1][y] == 1:
        y+=1
    else:
        y+=1

for i in board:
    print(' '.join(repr(e) for e in i))

---

입력 예시 :

1 1 1 1 1 1 1 1 1 1

1 0 0 1 0 0 0 0 0 1

1 0 0 1 1 1 0 0 0 1

1 0 0 0 0 0 0 1 0 1

1 0 0 0 0 0 0 1 0 1

1 0 0 0 0 1 0 1 0 1

1 0 0 0 0 1 2 1 0 1

1 0 0 0 0 1 0 0 0 1

1 0 0 0 0 0 0 0 0 1

1 1 1 1 1 1 1 1 1 1

 

출력 예시:

1 1 1 1 1 1 1 1 1 1

1 9 9 1 0 0 0 0 0 1

1 0 9 1 1 1 0 0 0 1

1 0 9 9 9 9 9 1 0 1

1 0 0 0 0 0 9 1 0 1

1 0 0 0 0 1 9 1 0 1

1 0 0 0 0 1 9 1 0 1

1 0 0 0 0 1 0 0 0 1

1 0 0 0 0 0 0 0 0 1

1 1 1 1 1 1 1 1 1 1