[ 백준 / # 17143 / Gold I] 낚시왕 (Java)

문제 설명

https://www.acmicpc.net/problem/17143

17143번: 낚시왕

[Gold I] 낚시왕 - 17143

성능 요약

메모리: 25904 KB, 시간: 2448 ms

분류

구현, 시뮬레이션

제출 일자

2024년 4월 11일 13:33:00

문제 설명

낚시왕이 상어 낚시를 하는 곳은 크기가 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인 경우는 왼쪽을 의미한다.

두 상어가 같은 크기를 갖는 경우는 없고, 하나의 칸에 둘 이상의 상어가 있는 경우는 없다.

출력

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

 

 

---

문제풀이

복잡한 구현문제이지만, 역시 난 구현문제가 제일 재밌다.

상어를 클래스로 만들어서 해시맵에 상어의 인덱스를 키로주고 상어 객체를 value로 주었다.

먼저 낚시왕이 상어를 잡는다

catchShark 메서드
자신이 속한 열에서 가장 땅에 가까운(위쪽에 있는) 상어를 잡고 그 상어의 크기를 답에 더함
잡은 상어는 해시맵에서 삭제

그 다음 상어들이 움직인다.
moveShark 메서드
해시맵의 keySet을 받아와 살아있는 모든 상어들을 움직인다.
상어의 각 속력만큼 움직이는데 속한 영역을 벗어나지 못하므로 가장자리에 다다르면 반대 방향으로 더 움직인다.
상어가 움직임을 멈췄을 때 그 칸에 다른 상어가 있다면 더 큰 상어가 작은 상어를 잡아먹는다.
잡아 먹힌 상어를 해시맵에서 삭제한다.

낚시 왕이 1열부터 끝 열까지 낚시를 끝내면 답을 출력하고 종료한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.StringTokenizer;

class Shark {
    int r, c, s, d, z;

    public Shark(int r, int c, int s, int d, int z) {
        this.r = r;
        this.c = c;
        this.s = s;
        this.d = d;
        this.z = z;
    }

}

public class bj_17143 {
    static int r, c, m, map[][], ans;
    static int[][] dir = { { -1, 0 }, { 1, 0 }, { 0, 1 }, { 0, -1 } };
    static HashMap<Integer, Shark> sharks;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        r = Integer.parseInt(st.nextToken());
        c = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        sharks = new HashMap<>();
        map = new int[r + 1][c + 1];
        for (int i = 1; i <= m; i++) {
            st = new StringTokenizer(br.readLine());
            sharks.put(i, new Shark(Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()),
                    Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()) - 1,
                    Integer.parseInt(st.nextToken())));

            map[sharks.get(i).r][sharks.get(i).c] = i;
        }

        for (int i = 1; i <= c; i++) {
            chatchShark(i);
            moveShark();
        }
        System.out.println(ans);
    }

    public static void chatchShark(int c) {
        for (int i = 1; i <= r; i++) {
            if (map[i][c] != 0) {
                Shark s = sharks.get(map[i][c]);
                ans += s.z;
                sharks.remove(map[i][c]);
                map[i][c] = 0;
                return;
            }
        }
    }

    public static void moveShark() {
        int[][] newMap = new int[r + 1][c + 1];

        for (int index : new ArrayList<>(sharks.keySet())) {

            Shark s = sharks.get(index);
            int nr = s.r;
            int nc = s.c;
            for (int i = 0; i < s.s; i++) {
                if (nr + dir[s.d][0] > 0 && nr + dir[s.d][0] <= r && nc + dir[s.d][1] > 0
                        && nc + dir[s.d][1] <= c) {
                    nr += dir[s.d][0];
                    nc += dir[s.d][1];
                } else {
                    switch (s.d) {
                        case 0:
                            s.d = 1;
                            nr += dir[s.d][0];
                            nc += dir[s.d][1];
                            break;
                        case 1:
                            s.d = 0;
                            nr += dir[s.d][0];
                            nc += dir[s.d][1];
                            break;
                        case 2:
                            s.d = 3;
                            nr += dir[s.d][0];
                            nc += dir[s.d][1];
                            break;
                        case 3:
                            s.d = 2;
                            nr += dir[s.d][0];
                            nc += dir[s.d][1];
                            break;
                    }
                }
            }
            s.r = nr;
            s.c = nc;
            if (newMap[nr][nc] == 0) {
                newMap[nr][nc] = index;
            } else {
                if (sharks.get(newMap[nr][nc]).z < sharks.get(index).z) {
                    sharks.remove(newMap[nr][nc]);
                    newMap[nr][nc] = index;
                } else {
                    sharks.remove(index);
                }

            }
        }
        map = newMap;
    }
}