[Baekjoon] 21610번 : 마법사 상어와 비바라기 - Python

Gold 5

21610번: 마법사 상어와 비바라기

 

 

문제 풀이

구현 문제

문제에서 제시한 조건대로 차례차례 구현하면 되는 문제였다.

cloud 리스트를 따로 만들어 구름의 위치를 다룰 수 있게 했다.

 

코드

from copy import deepcopy

n, m = map(int, input().split())
map_ = []
for _ in range(n):
    map_.append(list(map(int, input().split())))
cloud = [[0] * n for _ in range(n)]


#  ←, ↖, ↑, ↗, →, ↘, ↓, ↙
mx = [0, -1, -1, -1, 0, 1, 1, 1]
my = [-1, -1, 0, 1, 1, 1, 0, -1]

def move_cloud(direction, x, y, dist):
    temp_x, temp_y = x, y

    for _ in range(dist):
        nx = mx[direction] + temp_x
        ny = my[direction] + temp_y

        # 범위 밖을 벗어날 경우 처리
        nx = n-1 if nx < 0 else nx
        ny = n-1 if ny < 0 else ny
        nx = 0 if nx >= n else nx
        ny = 0 if ny >= n else ny

        temp_x, temp_y = nx, ny

    return nx, ny

# 비바라기 시전
cloud[n-1][0], cloud[n-1][1], cloud[n-2][0], cloud[n-2][1] = 1, 1, 1, 1
count = 0

while True:
    if count == m:
        sum_ = 0
        for i in range(n):
            sum_ += sum(map_[i])
        print(sum_)
        break

    direction, distance = map(int, input().split())
    temp_cloud = [[0] * n for _ in range(n)]

    # 구름 이동
    for i in range(n):
        for j in range(n):
            if cloud[i][j] == 1:
                move_x, move_y = move_cloud(direction-1, i, j, distance)
                temp_cloud[move_x][move_y] = 1
                map_[move_x][move_y] += 1

    cloud = deepcopy(temp_cloud)

    # 물복사버그 마법 시전
    for i in range(n):
        for j in range(n):
            if cloud[i][j]:
                bucket = 0
                for k in range(4):
                    copy_direction = k*2 + 1
                    nx = i + mx[copy_direction]
                    ny = j + my[copy_direction]

                    if 0 <= nx < n and 0 <= ny < n and map_[nx][ny]:
                        bucket += 1
                # 물이 있는 바구니 수 만큼 물 증가
                map_[i][j] += bucket

    # 바구니에 저장된 물의 양이 2 이상인 모든 칸에 구름 생성
    # 물의 양 -2
    for i in range(n):
        for j in range(n):
            # 구름이 있는 칸은 구름 삭제
            if cloud[i][j]:
                cloud[i][j] = 0

            # 구름이 없으면서 2이상이면 구름이 생기고 물의 양 2 감소
            else :
                if map_[i][j] >= 2:
                    cloud[i][j] = 1
                    map_[i][j] -= 2

    count += 1

 

느낀점

1시간 40분 정도 소요

예전보단 빨라졌지만 어렵지 않은데 비해 잔 실수들 때문에 찾는데 시간이 좀 오래걸렸다 ㅎ.. 오타나 실수를 줄일 수 있도록 하자.

 

범위 밖을 벗어날 때를 이어가는 유형은 처음 봤는데 그 점이 지금까지 접해왔던 문제들과는 좀 달랐던 것 같다.

이런 경우에 이동하는 부분은 아래처럼 나머지를 계산 하는것이 더 깔끔하고 효율적이다.

nx = (x + mx[direction-1]*distance) % n
ny = (y + my[direction-1]*distance) % n