[백준 / BAEKJOON] 17143 낚시왕 - python

문제 원문

조건

시간 제한 : 1초
메모리 제한 : 512 MB

문제

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

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

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

상어는 입력으로 주어진 속도로 이동하고, 속도의 단위는 칸/초이다. 상어가 이동하려고 하는 칸이 격자판의 경계를 넘는 경우에는 방향을 반대로 바꿔서 속력을 유지한채로 이동한다.

왼쪽 그림의 상태에서 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. 낚시왕이 오른쪽으로 한 칸 이동한다.
2. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다.
3. 상어가 이동한다.
4. 한 위치에 여러 상어가 있을 경우 크기가 큰 상어가 작은 상어를 먹는다

이 4가지 동작 중에 상어를 이동하는 기능과 한 위치에 여러 상어가 있을 경우 조건에 따라 처리하는 기능입니다.

저는 상어가 이동하는 기능은 상어의 정보와 기능을 가지고 있는 클래스를 작성하여 내부 메소드로 구현하였습니다.

한 위치에 여러 상어가 있을 때 처리는 이동한 결과를 담는 배열을 새로 생성하여 처리하였습니다. 만약 상어가 이동한 위치에 상어가 이미 있다면 그것은 이동을 완료한 상어이고 크기가 작은 상어는 잡아 먹힙니다.

때문에 이동 후 배열에 입력할 때 이미 상어가 저장되어 있다면 크기를 비교해 더 큰 상어를 저장하도록 하였고 먹힌 상어는 죽은 상어로 따로 배열에 저장해두어 이동이 모두 완료 되었을 때 상어의 정보를 담고있는 리스트에서 제거하였습니다.

풀이과정은 다음과 같습니다.

1. 상어의 정보와 기능을 가진 클래스를 선언한다
   1. 상어의 위치(x, y), 속도, 방향, 크기 정보를 저장하고 있다
   2. 속도가 왕복하는 거리보다 크면 의미없는 계산이므로 speed % ((boardSize - 1) * 2) 값으로 저장한다
   3. move 메소드는 방향에 따라 다르게 동작하며 이동 거리가 영역을 벗어날 경우 영역의 끝에 도달한 것으로 계산한 다음에 방향을 바꾸어 나머지 이동거리 만큼 다시 이동한다.
2. 격자판에 상어를 저장하고 상어 정보를 배열에 저장한ㄷ
3. 낚시왕이 영역을 벗어날 때 까지 반복한다. 단, 설명에선 움직인 후 낚시를 하였지만 이 풀이에선 처음부터 낚을 수 있는 위치에서 시작하여 낚시를 하고 이동하는 것으로 진행하였다.
4. 낚시왕이 낚시를 하고 낚였으면 해당 상어를 리스트에서 제거한다. 낚인 상어의 크기만큼 결과값에 더해준다
5. 상어가 이동하고 만약 중복됐을 경우 따로 배열에 죽은 상어를 저장한다
6. 죽은 상어를 리스트에서 제거한다
7. 낚시왕이 이동한다

성공 코드

import string
import sys
import heapq
from types import resolve_bases

moveX = [0, -1, 1, 0, 0]
moveY = [0, 0, 0, 1, -1]

sys.setrecursionlimit(10**9)
input = sys.stdin.readline

boardSizeX, boardSizeY, sharkTotalCount = map(int, input().split())

def dictReverse(dict):
    result = [0, 2, 1, 4, 3]
    return result[dict]

class Shark:
    def __init__(self, x, y, s, d, z):
        self.x = x
        self.y = y
        self.__dict = d
        self.__size = z

        #speed가 왕복하는 거리보다 크면 나머지값만 계산한다
        if self.__dict == 1 or self.__dict == 2:
            self.__speed = s % ((boardSizeX - 1) * 2)
        else:
            self.__speed = s % ((boardSizeY - 1) * 2)

    def move(self):
        moveCount = self.__speed

        #방향에 따라 0 또는 (size - 1)까지의 거리보다 높으면 해당 거리로 이동 후 다시 계산
        while moveCount:
            if self.__dict == 1: #위
                if self.x < moveCount:
                    moveCount -= self.x
                    self.x = 0
                    self.__dict = dictReverse(self.__dict)
                else:
                    self.x -= moveCount
                    moveCount = 0
            elif self.__dict == 2: #아래
                if self.x + moveCount >= boardSizeX:
                    moveCount -= (boardSizeX - 1) - self.x
                    self.x = boardSizeX - 1
                    self.__dict = dictReverse(self.__dict)
                else:
                    self.x += moveCount
                    moveCount = 0
            elif self.__dict == 3: #오른쪽
                if self.y + moveCount >= boardSizeY:
                    moveCount -= (boardSizeY - 1) - self.y
                    self.y = boardSizeY - 1
                    self.__dict = dictReverse(self.__dict)
                else:
                    self.y += moveCount
                    moveCount = 0        
            if self.__dict == 4: #왼쪽
                if self.y < moveCount:
                    moveCount -= self.y
                    self.y = 0
                    self.__dict = dictReverse(self.__dict)
                else:
                    self.y -= moveCount
                    moveCount = 0

    def getSize(self):
        return self.__size


if sharkTotalCount == 0:
    print(0)
else:
    #입력
    sharkData = []
    board = [[None] * boardSizeY for _ in range(boardSizeX)]

    for _ in range(sharkTotalCount):
        x, y, speed, dict, size = map(int, input().split())
        x -= 1
        y -= 1
        sharkData.append(Shark(x, y, speed, dict, size))
        board[x][y] = sharkData[-1]

    fishingKingPos = 0
    answer = 0

    while fishingKingPos < boardSizeY:
        #1. 낚시왕이 있는 열에 있는 상어 중에서 땅과 제일 가까운 상어를 잡는다. 상어를 잡으면 격자판에서 잡은 상어가 사라진다.
        for x in range(boardSizeX):
            if board[x][fishingKingPos] != None:
                answer += board[x][fishingKingPos].getSize()
                #상어 리스트에서 제거
                sharkData.remove(board[x][fishingKingPos])
                board[x][fishingKingPos] = None           
                break
        #2. 상어가 이동한다.
        tempBoard = [[None] * boardSizeY for _ in range(boardSizeX)]
        deadShark = []
        for shark in sharkData:
            board[shark.x][shark.y] = None
            shark.move()
            
            #상어 중복 확인
            if tempBoard[shark.x][shark.y] != None:
                if shark.getSize() > tempBoard[shark.x][shark.y].getSize():
                    deadShark.append(tempBoard[shark.x][shark.y])
                    tempBoard[shark.x][shark.y] = shark
                else:
                    deadShark.append(shark)
            else:
                tempBoard[shark.x][shark.y] = shark

        board = tempBoard

        #죽은 상어 제거
        for shark in deadShark:
            sharkData.remove(shark)

        #3. 낚시왕이 오른쪽으로 한 칸 이동한다.
        fishingKingPos += 1

    print(answer)

결과

Github - github.com/ledpear