코드업 6098 성실한 개미 Python

리스트 안에 리스트를 넣은

(0,0)으로 시작하는 [x][y] 좌표의 ary를 만든다.

 

좌표 배열과 관련한 문제는 코딩 테스트의 단골이니, 익숙해지면 좋을 것 같다.

ary = []
for i in range(10):
    temp=list(map(int,input().split()))
    ary.append(temp)
x=1
y=1
ary[x][y] = 9 ## (2,2)지점에서 출발

for i in range(10):
    for j in range(10):
        if ary[i][j] == 1: ## 장애물
            ary[i][j] = 1
        elif ary[i][j] == 9 and ary[i][j+1] == 0: ## 우측 이동
            ary[i][j+1] = 9
        elif ary[i][j] == 9 and ary[i][j+1] == 1 and ary[i+1][j] == 0: ## 하단 이동
            ary[i+1][j] = 9

for i in range(10):
    for j in range(10):
        if ary[i][j] == 9 and ary[i][j+1] == 2: ## 우측에서 먹이 발견
            ary[i][j+1] = 9
        elif ary[i][j] == 9 and ary[i][j+1] == 1 and ary[i+1][j] == 2: ## 하단에서 먹이 발견
            ary[i+1][j] = 9


for m in range(10):
    for n in range(10):
        print(ary[m][n], end=" ")
    print()

* https://www.codeup.kr/problem.php?id=6098