코드업 기초 100제 문제 099

코드업 기초 100제  문제 099

 

 

# 성실한 개미

# **서론**

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

# 왕개미를 유심히 살펴보던 중 특별히 성실해 보이는 개미가 있었는데,

# 그 개미는 개미굴에서 나와 먹이까지 가장 빠른 길로 이동하는 것이었다.

# 개미는 오른쪽으로 움직이다가 벽을 만나면 아래쪽으로 움직여 가장 빠른 길로 움직였다.

# (오른쪽에 길이 나타나면 다시 오른쪽으로 움직인다.)

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

# 미로 상자에 넣은 개미는 먹이를 찾았거나, 더 이상 움직일 수 없을 때까지

# 오른쪽 또는 아래쪽으로만 움직였다.

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

---

소스

# 입력
board = []
for _ in range(10):
    board.append(list(map(int, input().split())))
x = y = 1
board2 = board1 = board
print('===================')
# 1
d = 1
while board1[x][y] != 2:
    board1[x][y] = 9
    if d:  # 방향이 오른쪽
        if board1[x][y+1] == 1:  # 오른쪽 검사
            d = 0
        else:
            y += 1
    else:  # 방향이 아래쪽
        if board1[x+1][y] == 1:  # 아래쪽 검사
            break
        x += 1
        d = 1
board1[x][y] = 9
for brd in board1:
    print(*brd)
print('===================')
# 2
while board2[x][y] != 2:
    board2[x][y] = 9
    if board2[x][y+1] != 1:  # 오른쪽이 안막힌 경우
        y += 1
    elif board2[x+1][y] != 1:  # 아래쪽이 안막힌 경우
        x += 1
    else:  # 둘다 막힌 경우
        break
board2[x][y] = 9
for brd in board2:
    print(*brd)

이문제에서 개미는 오른쪽으로 가다가 길이 막힐 경우 아래쪽으로 움직이고 아래쪽으로 가던 중 오른쪽에 길이 나오면 다신 오른쪽으로 간다고 합니다. 그리고 개미가 멈출 때는 먹이를 찾거나 더 이상 움직일 수 없을 때입니다.

1번 풀이 같은 경우 방향이라는 변수를 줘서 오른쪽을갈 때와 아래쪽으로 갈때로 소스를 나눕니다. 방향이 오른쪽으로 갈때 길 앞에 장애물이 있는지 검사하여 있으면 방향을 다시 바꿔 반복문을 돌립니다. 없으면 오른쪽으로 갑니다.  아래쪽으로 갈 때도 마찬가지로 검사를 해줍니다. 대신 여기서는 한번 가고 나면 방향을 오르 쪽으로 바꿉니다. 이렇게 해서 아래쪽을 검사를 해서 막히게 되면 오른쪽과 아래쪽 둘 다 막힌 것을 의미하게 되므로 break로 반복문에서 빠져나올 수 있게 됩니다.

 

2번 풀이는 1번 풀이를 줄인 결과입니다. 매번 검사를 할 때 오른쪽 이후 아래쪽을 검사하게 됩니다. 이것은 if~ elif~else~문의 검사 순서와도 비슷하기 때문에 위 과정을 if~ elif~else~문에 이식을 해주면 위 과정의 소스보다 더욱 간단하게 소스를 짤 수 있게 됩니다.