[Algorithm] 백준/17143 :: 낚시왕

* Review

2019 상반기 공채 오전반 복원문제중 2번문제였다.

기본 삼성유형으로 알려진 BFS, DFS에서 그냥 구현문제가 나올줄은 몰랏지만

쉬웠다는 평이 많은 상반기 코테에서 1솔로 가능할지 의문이다.

* 핵심 개념

이 문제에서는 주의할 점이 두가지 있다.

1. 물고기들 각각을 움직일 때 바로 움직이지 않고, 한번에 움직여야 한다.

필자는 Fish 배열을 만들어 정보를 저장하고 상어 움직임이 끝낫을 때 한번에 이동했다.

물론, 조건에 따라 해당 좌표에 자신보다 작은 상어가 있다면 Fish 배열에 null로 바꿔주면 된다.

2. 시간 속도를 줄이려면 움직여야 하는 거리 % (Width or Depth * 2 - 2) 계산을 하고 움직이면 된다.(584ms -> 308ms)

아쉽게도 시험장에서 30분을 고민하다 이 공식을 못찾았다.

이 두가지 조건만 만족하면 어렵지 않게 풀수있는 기출문제다.

* 문제

문제

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

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

출력

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

* 소스 코드

import java.io.*;
import java.util.*;
 
public class Main {
    public static StringTokenizer stk;
    public static StringBuilder sb = new StringBuilder();
    public static int[] dy = {-1, 1, 0, 0};
    public static int[] dx = {0, 0, 1, -1};
    public static int width, depth, n, ans = 0;
    public static Fish[] fishInfo = new Fish[10001];
    public static int[][] map;
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        stk = new StringTokenizer(br.readLine());
        depth = Integer.parseInt(stk.nextToken());
        width = Integer.parseInt(stk.nextToken());
        n = Integer.parseInt(stk.nextToken());
        map = new int[depth + 1][width + 1];
        for (int i = 1; i <= n; i++) {
            stk = new StringTokenizer(br.readLine());
            int r = Integer.parseInt(stk.nextToken());
            int c = Integer.parseInt(stk.nextToken());
            int s = Integer.parseInt(stk.nextToken());
            int d = Integer.parseInt(stk.nextToken());
            int z = Integer.parseInt(stk.nextToken());
            fishInfo[z] = new Fish(z, c, r, d - 1, s);
            map[r][c] = z;
        }
        for (int i = 1; i <= width; i++) {
            //작살로 물고기 잡음
            for (int j = 1; j <= depth; j++) {
                if (map[j][i] != 0) {
                    fishInfo[map[j][i]] = null;
                    ans += map[j][i];
                    map[j][i] = 0;
                    break;
                }
            }
            //물고기 이동
            for (int j = 1; j <= 10000; j++) {
                if (fishInfo[j] == null) continue;
                Fish curr = fishInfo[j];
                int nx = curr.x;
                int ny = curr.y;
                int move = curr.speed;
                int d = curr.d;
                map[ny][nx] = 0;
                switch (d) {    //방향따라 case 다르게
                    //상하
                    case 0:
                    case 1:
                        move %= (depth * 2 - 2);
                        while (move > 0) {
                            if (ny == 1) {
                                d = 1;
                            }
                            if (ny == depth) {
                                d = 0;
                            }
                            ny += dy[d];
                            move--;
                        }
                        fishInfo[curr.idx] = new Fish(curr.idx, nx, ny, d, curr.speed);
                        break;
                    //좌우
                    case 2:
                    case 3:
                        move %= (width * 2 - 2);
                        while (move > 0) {
                            if (nx == 1) d = 2;
                            if (nx == width) d = 3;
                            nx += dx[d];
                            move--;
                        }
                        fishInfo[curr.idx] = new Fish(curr.idx, nx, ny, d, curr.speed);
                        break;
                }
            }
            for (int j = 1; j <= 10000; j++) {
                if (fishInfo[j] == null) continue;
                Fish curr = fishInfo[j];
                if (map[curr.y][curr.x] < curr.idx) {   //맵에 저장된 물고기보다 본인이 더 크면
                    fishInfo[map[curr.y][curr.x]] = null;
                    map[curr.y][curr.x] = curr.idx;
                }
            }
        }
        System.out.println(ans);
    }
 
    public static class Fish {
        int idx, x, y, d, speed;
 
        public Fish(int idx, int x, int y, int d, int speed) {
            this.idx = idx;
            this.x = x;
            this.y = y;
            this.d = d;
            this.speed = speed;
        }
    }
}