[BOJ][🟡1][백준#17143] 낚시왕

문제 출처

BAEKJOON Online Judge #17143

문제

낚시왕이 상어 낚시를 하는 곳은 크기가 R×C인 격자판으로 나타낼 수 있다. 격자판의 각 칸은 (r, c)로 나타낼 수 있다. r은 행, c는 열이고, (R, C)는 아래 그림에서 가장 오른쪽 아래에 있는 칸이다. 칸에는 상어가 최대 한 마리 들어있을 수 있다. 상어는 크기와 속도를 가지고 있다.

낚시왕은 처음에 1번 열의 한 칸 왼쪽에 있다. 다음은 1초 동안 일어나는 일이며, 아래 적힌 순서대로 일어난다. 낚시왕은 가장 오른쪽 열의 오른쪽 칸에 이동하면 이동을 멈춘다.

낚시왕이 오른쪽으로 한 칸 이동한다. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다. 상어가 이동한다.

상어는 입력으로 주어진 속도로 이동하고, 속도의 단위는 칸/초이다. 상어가 이동하려고 하는 칸이 격자판의 경계를 넘는 경우에는 방향을 반대로 바꿔서 속력을 유지한채로 이동한다. 왼쪽 그림의 상태에서 1초가 지나면 오른쪽 상태가 된다. 상어가 보고 있는 방향이 속도의 방향, 왼쪽 아래에 적힌 정수는 속력이다. 왼쪽 위에 상어를 구분하기 위해 문자를 적었다.

상어가 이동을 마친 후에 한 칸에 상어가 두 마리 이상 있을 수 있다. 이때는 크기가 가장 큰 상어가 나머지 상어를 모두 잡아먹는다. 낚시왕이 상어 낚시를 하는 격자판의 상태가 주어졌을 때, 낚시왕이 잡은 상어 크기의 합을 구해보자.

입력

첫째 줄에 격자판의 크기 R, C와 상어의 수 M이 주어진다. (2 ≤ R, C ≤ 100, 0 ≤ M ≤ R×C) 둘째 줄부터 M개의 줄에 상어의 정보가 주어진다. 상어의 정보는 다섯 정수 r, c, s, d, z (1 ≤ r ≤ R, 1 ≤ c ≤ C, 0 ≤ s ≤ 1000, 1 ≤ d ≤ 4, 1 ≤ z ≤ 10000) 로 이루어져 있다. (r, c)는 상어의 위치, s는 속력, d는 이동 방향, z는 크기이다. d가 1인 경우는 위, 2인 경우는 아래, 3인 경우는 오른쪽, 4인 경우는 왼쪽을 의미한다. 두 상어가 같은 크기를 갖는 경우는 없고, 하나의 칸에 둘 이상의 상어가 있는 경우는 없다.

출력

낚시왕이 잡은 상어 크기의 합을 출력한다.

예제

예제 1

입력

4 6 8
4 1 3 3 8
1 3 5 2 9
2 4 8 4 1
4 5 0 1 4
3 3 1 2 7
1 5 8 4 3
3 6 2 1 2
2 2 2 3 5

출력

22

예제 2

입력

100 100 0

출력

0

예제 3

입력

4 5 4
4 1 3 3 8
1 3 5 2 9
2 4 8 4 1
4 5 0 1 4

출력

22

예제 4

입력

2 2 4
1 1 1 1 1
2 2 2 2 2
1 2 1 2 3
2 1 2 1 4

출력

4

My Sol

import sys
input = sys.stdin.readline

def main():
    def place_sharks():
        sharks = {}
        for m in range(M):
            i, j, s, d, z = map(int, input().split())
            mat[i-1][j-1].append((z, s, d, m))
            sharks[m] = (i-1, j-1, z, s, d)
        return sharks

    def hunt_shark():
        size = 0
        for i in range(I):
            if not mat[i][kj]: continue
            z, s, d, m = mat[i][kj].pop()
            sharks[m] = ()
            sharks_nums.remove(m)
            size = z
            break
        return size

    def move_sharks():
        new_sharks = {}
        for m in sharks_nums:
            i, j, z, s, d = sharks[m]
            mat[i][j].pop()
            ni, nj, nd = move_shark(i, j, s, d)
            new_sharks[m] = (ni, nj, z, s, nd)
        return new_sharks

    def move_shark(i, j, s, d):
        if not s:
            return i, j, d

        di, dj = DD[d]
        if di:
            return move_vertical(i, j, s, d)
        else:
            return move_horizontal(i, j, s, d)

    def move_vertical(i, j, s, d):
        di, dj = DD[d]
        s = s%((I-1)*2)
        while True:
            si = i+di*s
            if 0<=si<I: return si, j, d
            move_i = I-1-i if di > 0 else i
            i = i+move_i if di > 0 else i-move_i
            s -= move_i
            di *= -1
            d = RD[d]

    def move_horizontal(i, j, s, d):
        di, dj = DD[d]
        s = s%((J-1)*2)
        while True:
            sj = j+dj*s
            if 0<=sj<J: return i, sj, d
            move_j = J-1-j if dj > 0 else j
            j = j+move_j if dj > 0 else j-move_j
            s -= move_j
            dj *= -1
            d = RD[d]

    def fill_sharks():
        for m in sharks_nums:
            i, j, z, s, d = sharks[m]
            if not mat[i][j]:
                mat[i][j].append((z, s, d, m))

            else:
                if mat[i][j][0][0] > z:
                    remove_sharks_set.add(m)
                else:
                    rm = mat[i][j][0][3]
                    remove_sharks_set.add(rm)
                    mat[i][j].pop()
                    mat[i][j].append((z, s, d, m))

    def remove_sharks():
        for rm in remove_sharks_set:
            remove_shark(rm)
        remove_sharks_set.clear()

    def remove_shark(m):
        sharks_nums.remove(m)
        sharks[m] = ()


    I, J, M = map(int, input().split())
    mat = [[[] for _ in range(J)] for _ in range(I)]
    sharks = place_sharks()
    sharks_nums = set(range(M))
    remove_sharks_set = set()

    RD = {1:2, 2:1, 3:4, 4:3}
    DD = {1:(-1,0), 2:(1,0), 3:(0,1), 4:(0,-1)}

    ret = 0
    for kj in range(J):
        ret += hunt_shark()
        sharks = move_sharks()
        fill_sharks()
        remove_sharks()
    return ret

print(main())

꽤나 복잡했던 구현 문제였다. 하지만 단계별로 잘 모듈화해서 파훼한다면 어렵지 않게 풀어낼 수 있었다.

1. 입력을 받는다. 상어들의 초기 위치를 받는 place_sharks 함수를 작성해 mat 2차원 배열에 상어 정보를 배치함은 물론, 상어 번호들을 지정한 sharks set을 두었다. 이 sharks set의 정보들을 바탕으로 후에 남은 shark들의 이동을 매번 해주게 될 것이다.
2. 낚시꾼이 오른쪽으로 초당 이동하는 것을 for문을오 작성하였다. 각 열마다 가장 위에 있는 상어를 제거하고 그 크기를 반환하는 함수 hunt_shark 함수를 작성하였고, 이를 최종 결과값인 ret에 더해주었다.
3. 이후 각 상어마다 움직여주는 함수 move_sharks 함수를 작성하였는데, 상어마다 방향을 바꾸고 정보를 재조정해주어야 하기 때문에 상어당 이동을 하는 move_shark 함수를 작성하였고, 이는 다시 이동 방향에 따라 move_vertical과 move_horizontal로 나누어 범위를 벗어났는지 확인하며 최종 이동 좌표를 확정한다.
4. 상어들의 각 이동 후 정보들을 계산한 뒤 fill_sharks 함수를 통해 비워진 mat 2차원 배열에 넣어주었다. 이 과정에서 상어가 겹친다면 크기가 큰 상어만 남겨주어야 했다. 그런데 set을 순회하는 과정에서 set의 크기가 달라지면 안 됐기 때문에 main에 remove_sharks_set set을 두어서 제거해야 하는 상어 번호를 넣어주었다.
5. 이후 remove_sharks 함수를 통해 제거 대상 상어의 정보를 sharks에서 지웠다.
6. 모든 과정에서 sharks dictionary와 sharks_nums set이 있는데, 처음 상어의 각 정보마다 번호를 매겨 제거되거나 사냥당한 상어는 정보를 지워주는 등의 과정을 거친다. 이 과정에서 편의를 위해 index 참조 dictionary와 set을 따로 둔 것이다.

결과

맞았습니다!!