[코드업 문제풀이] 6098 : [기초-리스트] 성실한 개미

코드업 파이썬 기초 알고리즘 문제 100제 중에서 6098번에 대한 풀이이다.

100제 끝.

import sys
d = [[int(x) for x in sys.stdin.readline().rstrip().split()] for y in range(10)]
d[1][1] = 9

i = 1
j = 1
while j < 9:
    if d[i][j+1] == 0:
        d[i][j+1] = 9
        j += 1
    elif d[i][j+1] == 1:
        if d[i+1][j] == 1:
            break
        elif d[i+1][j] == 2:
            d[i+1][j] = 9
            break
        else:
            d[i + 1][j] = 9
            i += 1
    elif d[i][j+1] == 2:
        d[i][j+1] = 9
        break;
    else:
        d[j][i+1] = 9
        i += 1

for i in d:
    for j in i:
        print(j, end=' ')
    print()