Codeup이란?

코드업은 국내의 한 정보 교사가 알고리즘 교육을 목적으로 운영하는 사이트입니다.
난이도가 낮고 문제가 많아 처음 코딩 테스트를 입문하려는 사람들에게 좋습니다.

문제가 쉽다고 무시하지 말고 코딩테스트는 어떤 형태로 나오는지, 문법은 무엇인지 연습하기 좋으며 쉬운 문제를 풀어가면서 문제풀이에대한 자신감을 키우기 좋기 때문에 입문자는 코드업 사이트의 [홈] - [기초100제] - [Python 기초100제]를 꼭 풀어보도록 합시다.

코드업 Python 기초 100제로 이동

 

문제

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

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

 

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

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

 

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

 

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

 

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

 

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

 

입력

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

예시

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

 

출력

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

예시

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

 

내 답안

Click to expand!!
d = [[0 for _ in range(11)] for _ in range(11)]
for i in range(1, 11):
  n = map(int, input().split())
  for (j, value) in enumerate(n):
    d[i][j+1] = value

x, y = 2, 2
if d[y][x] == 2:
  d[y][x] = 9

else:
  t = d[y][x]
  while t != 2:
    d[y][x] = 9
    if d[y][x+1] != 1:
      x += 1
    elif d[y+1][x] != 1:
      y += 1
    else:
      break
    t = d[y][x]
  d[y][x] = 9
    
for i in range(1, 11):
  for j in range(1, 11):
    print(d[i][j], end=" ")
  print()