[코드업] 파이썬 기초 100제 6098번 | 성실한 개미

문제 설명

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

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

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

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

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

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

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

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

 

입력

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

 

출력

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

 

 

풀이

list = [list(map(int, input().split())) for _ in range(10)]

x=1
y=1
list[x][y]=9
while(True):
    if list[x][y+1]!=1:
        y += 1
    else:
        x += 1
    list[x][y] = 9
        
    if list[x+1][y]==2 and list[x][y+1]==1:
        list[x+1][y]=9
        break
    if list[x][y+1]==2:
        list[x][y+1]=9
        break
    
    if list[x+1][y]==1 and list[x][y+1]==1:
        break
    
for i in range(10):
    for j in range(10):
        print(list[i][j], end=' ')
    print()

 

list 변수에 리스트 10번 반복해서 입력받았다.

왼쪽 맨 위 칸을 (0,0)으로 두고 풀었다.

while 반복문의 내용은 다음과 같다.

• 오른쪽이 벽(1)이 아니면 y를 증가, 그렇지 않으면 x를 증가시킨 후 list[x][y]를 9로 변환
• 아래에 먹이(2)가 있고 오른쪽이 벽인 경우 x를 증가시킨 후 list[x][y]를 9로 변환 -> 종료
• 오른쪽에 먹이가 있는 경우 y를 증가시킨 후 list[x][y]를 9로 변환 -> 종료
• 오른쪽과 아래가 모두 벽인 경우 종료

 

🙋‍♂️코드업 풀이

m=[]
for i in range(12) :
  m.append([])
  for j in range(12) :
    m[i].append(0)

for i in range(10) :
  a=input().split()
  for j in range(10) :
    m[i+1][j+1]=int(a[j])

x = 2
y = 2
while True :
  if m[x][y] == 0 :
    m[x][y] = 9
  elif m[x][y] == 2 :
    m[x][y] = 9
    break

  if (m[x][y+1]==1 and m[x+1][y]==1) or (x==9 and y==9) :
    break

  if m[x][y+1] != 1 :
    y += 1
  elif m[x+1][y] != 1 :
    x += 1
    

for i in range(1, 11) :
  for j in range(1, 11) :
    print(m[i][j], end=' ')
  print()

---

코드업 풀이가 확실히 눈에 깔끔하게 들어온다. 너무 생각나는 대로 풀었나🤦‍♂️ 문제에서 (2,2)부터 시작이라 했으면 코드에서도 맞춰주는 게 좋은 건가?? 아직 뭐가 뭔지 아무것도 모르겠다ㅎ

 

하다 보면 알게 되겠지😅

 

코드업 기초 100제는 끝끝

 

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