6098 : [기초-리스트] 성실한 개미(py)

문제 설명

본 문제는 python 의 빠른 기초 학습을 위해 설계된 문제로서 python 코드 제출을 기준으로 설명되어 있습니다. 
------

영일이는 생명과학에 관심이 생겨 왕개미를 연구하고 있었다.

왕개미를 유심히 살펴보던 중 특별히 성실해 보이는 개미가 있었는데,
그 개미는 개미굴에서 나와 먹이까지 가장 빠른 길로 이동하는 것이었다.

개미는 오른쪽으로 움직이다가 벽을 만나면 아래쪽으로 움직여 가장 빠른 길로 움직였다.
(오른쪽에 길이 나타나면 다시 오른쪽으로 움직인다.)

이에 호기심이 생긴 영일이는 그 개미를 미로 상자에 넣고 살펴보기 시작하였다.

미로 상자에 넣은 개미는 먹이를 찾았거나, 더 이상 움직일 수 없을 때까지
오른쪽 또는 아래쪽으로만 움직였다.

미로 상자의 구조가 0(갈 수 있는 곳), 1(벽 또는 장애물)로 주어지고,
먹이가 2로 주어질 때, 성실한 개미의 이동 경로를 예상해보자.

단, 맨 아래의 가장 오른쪽에 도착한 경우, 더 이상 움직일 수 없는 경우, 먹이를 찾은 경우에는
더이상 이동하지 않고 그 곳에 머무른다고 가정한다.

미로 상자의 테두리는 모두 벽으로 되어 있으며,
개미집은 반드시 (2, 2)에 존재하기 때문에 개미는 (2, 2)에서 출발한다.

입력

10*10 크기의 미로 상자의 구조와 먹이의 위치가 입력된다.

출력

성실한 개미가 이동한 경로를 9로 표시해 출력한다.

입력 예시   예시 복사

 

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

---

< 내 풀이 >

box = []

for i in range(10):                                                  #10x10 박스를 만듦
    box.append([])
    for j in range(10):
        box[i].append(0)

for i in range(10):                                                  #사이트에 있는 코드 복사해서 입력받음
    b = input().split()                                              #줄바꿈, i행에 한 줄씩
    for j in range(10):
        box[i][j] = int(b[j])                                        #b의 j번째 값을 box i행의 j번째에 할당

x = 1
y = 1                                                                #1,1부터 시작 사람 눈엔 2,2
while True:
    if box[x][y] == 0:                                               #x,y좌표의 값이 0이면 9로 재할당
        box[x][y] = 9
    elif box[x][y] == 2:                                             #x,y좌표의 값이 2이면, 먹이. 9로 재할당 후 반복문 종료
        box[x][y] = 9
        break;
    if (box[x][y+1] == 1 and box[x+1][y]) == 1 or (x==9 and y==9):   #y, x좌표 다음이 모두 1이거나, x, y 모두 9이면
        break;                                                       # 반복문 종료
    if box[x][y+1] != 1 :                                            #y좌표 다음의 값이 0일 경우
        y += 1                                                       #y에 1 추가함으로 다음 좌표로 시작을 알림
    elif box[x+1][y] != 1 :                                          #x좌표 다음의 값이 0일 경우
        x += 1                                                       #1 추가 함으로 다음 좌표에서 시작을 알림
for i in range(10):                                                  #반복문으로 바뀐 박스 출력
    for j in range(10):
        print(box[i][j],end=" ")
    print()