백준 17143번 낚시왕 [JAVA]

문제

낚시왕이 상어 낚시를 하는 곳은 크기가 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. 상어 이동

이라는 세 가지의 과정이 있습니다.

 

첫 번째로 for문을 이용해 낚시왕의 위치를 한 칸씩 이동해 준 뒤, sharkCatch() 메서드를 통해 해당 열에서 가장 가까운 위치에 있는 상어를 잡았습니다. 이동하면서 잡은 상어의 크기의 합을 담아둘 ans변수에 잡은 상어의 크기 값을 넣어주고 격자판에서 해당 상어를 제거해주었습니다. 마지막으로 상어들이 이동하는 sharkMove() 메서드에서 변경된 상어들의 위치를 일시적으로 담아두기 위해 격자판과 동일한 크기의 copy배열을 사용해 움직인 상어들의 위치를 찍어주었습니다. 해당 스피드만큼 상어가 이동한 후, 이동한 상어 위치에 다른 상어가 없다면 배열에 해당 상어 정보를 넣어주고, 움직인 상어가 기존의 있던 상어보다 크기가 큰 경우 기존에 있던 상어는 사라지고 움직인 상어의 정보를 배열에 넣어주었습니다. 반대로 이동한 상어가 기존의 있던 상어보다 크기가 작은 경우 그 상어는 해당 위치에 존재할 수 없으므로 상어 정보를 remove 큐에 넣어주었습니다. 모든 상어들의 이동이 끝난 후, 상어들의 정보가 담겨있는 list과 비교해서 제거될 상어들의 정보를 제거해 준 뒤, 원래 격자판인 map 배열에 상어들이 이동하고 난 뒤의 위치를 복사해 넣어주었습니다.

이 과정을 낚시왕이 왼쪽에서 오른쪽으로 이동할 때까지 반복하여 잡은 상어의 크기를 출력해주었습니다.

 

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
 
 
public class Main_BJ_17143_낚시왕 {
    
    static class info{
        int x,y,speed,dir,size;
 
        public info(int x, int y, int speed, int dir, int size) {
            super();
            this.x = x;
            this.y = y;
            this.speed = speed;
            this.dir = dir;
            this.size = size;
        }
    }
    
    static int R,C,M, ans=0;
    static int [][] map, copy;
    static int [] dx = {0,-1,1,0,0};
    static int [] dy = {0,0,0,1,-1};
    
    static ArrayList<info> shark = new ArrayList<>();
    static Queue<Integer> remove = new LinkedList<>();
    
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        
        st = new StringTokenizer(br.readLine());
        R = Integer.parseInt(st.nextToken()); //X
        C = Integer.parseInt(st.nextToken()); //Y
        M = Integer.parseInt(st.nextToken()); //상어 수
        
        map = new int[R+1][C+1];
        
        for(int i=0;i<M;i++) {
            st = new StringTokenizer(br.readLine());
            int r = Integer.parseInt(st.nextToken());
            int c = Integer.parseInt(st.nextToken());
            int s = Integer.parseInt(st.nextToken());
            int d = Integer.parseInt(st.nextToken());
            int z = Integer.parseInt(st.nextToken());
            
            shark.add(new info(r,c,s,d,z));
            map[r][c] = z;
        }
        
        
        for(int i=1;i<=C;i++) {
            //상어잡기
            sharkCatch(i);
            //상어움직이기
            sharkMove();    
        }
        System.out.println(ans);
 
    }
    
    public static void print() {
        for(int i=1;i<R+1;i++) {
            for(int j=1;j<C+1;j++) {
                System.out.print(map[i][j]+" ");
            }
            System.out.println();
        }
    }
    
    public static void sharkCatch(int y) {
        Loop:
        for(int x=1;x<R+1;x++) {
            if(map[x][y]!=0) {
                for(int i=0;i<shark.size();i++) {
                    if(shark.get(i).x==x && shark.get(i).y==y) {
                        ans+=shark.get(i).size;
                        map[x][y]=0;
                        shark.remove(i);
                        break Loop;
                    }
                }
            }
        }
    }
    
    public static void sharkMove() {
        
        copy = new int[R+1][C+1];
        
        for (int i=0;i<shark.size();i++) {
            info temp = shark.get(i);
            
            map[temp.x][temp.y] = 0;
            
            for (int j = 0; j < temp.speed ; j++) {// d==1 상, d==2 하, d==3 우, d==4 좌
 
                if (temp.dir == 1 && temp.x == 1) temp.dir = 2;
                else if (temp.dir == 2 && temp.x == R)temp.dir = 1;
                else if (temp.dir == 3 && temp.y == C)temp.dir = 4;
                else if (temp.dir == 4 && temp.y == 1)temp.dir = 3;
 
                temp.x += dx[temp.dir];
                temp.y += dy[temp.dir];
            } //상어 이동 끝
            
            if(copy[temp.x][temp.y]==0) {
                copy[temp.x][temp.y] = temp.size;
            }else if(copy[temp.x][temp.y]<temp.size) { //움직인 상어가 기존에 있던 상어보다 큰 경우
                remove.add(copy[temp.x][temp.y]);
                copy[temp.x][temp.y] = temp.size;
            }else {
                remove.add(temp.size);
            }
        }
        
        while(!remove.isEmpty()) {
            int temp = remove.poll();
            for(int i=0;i<shark.size();i++) {
                if(temp==shark.get(i).size) {
                    shark.remove(i); break;
                }
            }
        }
        map = copy;
    }
 
}