티스토리 뷰

<문제>

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

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

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

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

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

미로 상자의 구조가 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


<나의 풀이>

x = 1
y = 1
size = 10
table = []

# table 입력받기
for _ in range(size):
  array = list(map(int, input().split()))
  table.append(array)

table[x][y] = 9  #시작지점 9설정

while(1):

  if table[x][y+1] == 0:     #오른쪽이 0 
    y += 1
    table[x][y] = 9
    
  elif table[x][y+1] == 1:   #오른쪽이 1 
    if table[x+1][y] == 0:       #아래가 0 
      x += 1
      table[x][y] = 9
    elif table[x+1][y] == 1:     #아래가 1
      break  #오른쪽 아랫쪽 막힘
    else:
      table[x+1][y] = 9          #아래가 2
      break # 오른쪽 막히고 아랫쪽 2
      
  else:                      #오른쪽이 2 
    y += 1
    table[x][y] = 9
    break # 오른쪽 2


# 결과 출력
for i in range(size):
    for j in range(size):
        print(table[i][j], end=' ')
    print()


 

'코딩테스트 > 코드업 기본 100제' 카테고리의 다른 글

[코드업] 1420 : 3등 찾기 - python  (0) 2022.03.17
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
TAG
more
«   2025/01   »
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31
글 보관함