코드업(CodeUp) 기초100제 1099 성실한 개미 풀이 | Python-2차원 배열

벌써 마지막이네요!

1099번은 성실한 개미입니다

고쳐야할 사항이 있다면 언제든지 댓글 달아주세요:)

1099 : [기초-2차원배열] 성실한 개미

문제 설명

영일이는 생명과학에 관심이 생겨 왕개미를 연구하고 있었다.
왕개미를 유심히 살펴보던 중 특별히 성실해 보이는 개미가 있었는데,
그 개미는 개미굴에서 나와 먹이까지 가장 빠른 길로 이동하는 것이었다.
개미는 오른쪽으로 움직이다가 벽을 만나면 아래쪽으로 움직여 가장 빠른 길로 움직였다.
(오른쪽에 길이 나타나면 다시 오른쪽으로 움직인다.)

이에 호기심이 생긴 영일이는 그 개미를 미로 상자에 넣고 살펴보기 시작하였다.
미로 상자에 넣은 개미는 먹이를 찾았거나, 더 이상 움직일 수 없을 때까지
오른쪽 또는 아래쪽으로만 움직였다.
미로 상자의 구조가 0(갈 수 있는 곳), 1(벽 또는 장애물)로 주어지고,
먹이가 2로 주어질 때, 성실한 개미의 이동 경로를 예상해보자.

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

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

입력

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

출력

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

입력 예시

출력 예시

풀이

matrix=[[0 for i in range(10)]for j in range(10)]		#2차원 배열을 0으로 초기화
for i in range(10):
    matrix[i]=input().split()		#행을 차례대로 입력받음
x,y=0,1					#시작점 바로 앞으로 초기화
while True:				#오른쪽으로 한칸 이동
    if int(matrix[x][y+1])==0:		#오른쪽이 0일 경우
        y+=1
        matrix[x][y]=9 
    elif int(matrix[x][y+1])==1:
        if int(matrix[x+1][y])==0:	#오른쪽이 1이고 아래쪽은 0인 경우
            x+=1
            matrix[x][y]=9
        elif int(matrix[x+1][y])==1:	#오른쪽이 1이고 아래쪽도 0인 경우
            break
        else:				#오른쪽이 1이고 아래쪽은 2인 경우
            x+=1
            matrix[x][y]=9
            break
    elif int(matrix[x][y+1])==2:	#오른쪽이 2인 경우
        y+=1
        matrix[x][y]=9
        break
    if x==9 or y==9:		#끝까지 왔을 경우
        break
for i in range(10):		#2차원 배열 출력
    for j in range(10):
        print(matrix[i][j], end=" ")
    print()

*리스트 컴프리헨션*

변수명=[식 또는 값 for i in 범위]

ex1) 리스트 길이를 n으로 지정하고 0으로 초기화

list = [0 for i in range(n)]

ex2) 2차원 배열을 0으로 초기화

matrix = [[0 for col in range(n)] for row in range(n)]