[JAVA] BOJ 백준 17143 낚시왕

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

17143번: 낚시왕

문제

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

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

예제 출력 1 복사

22

각 칸의 왼쪽 아래에 적힌 수는 속력, 오른쪽 아래는 크기, 왼쪽 위는 상어를 구분하기 위한 문자이다. 오른쪽 위에 ❤️는 낚시왕이 잡은 물고기 표시이다.

초기 상태

1초

2초 (E번 상어는 B번에게 먹혔다)

3초

4초

5초

6초

전체 코드

import java.util.*;
import java.io.*;

class Shark{
	public 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 Main {
	static int R;
	static int C;
	static int M;
	static int result;
	static List<Shark> s;
	static int king;
	static Shark[][] sea;
	static int[] dx = {0, -1, 1, 0, 0};
	static int[] dy = {0, 0, 0, 1, -1}; // 입력 인덱스를 맞추려고
	
    public static void main(String[] args) throws Exception {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
        
        R = Integer.parseInt(st.nextToken());
        C = Integer.parseInt(st.nextToken());
        sea = new Shark[R][C];
        
        int T = Integer.parseInt(st.nextToken());
        s = new ArrayList<>();
        for(int i=0; i<T; i++) {
        	st = new StringTokenizer(bf.readLine(), " ");
        	int r = Integer.parseInt(st.nextToken()) -1;
        	int c = Integer.parseInt(st.nextToken()) -1;
        	Shark sk = new Shark(r, c, Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()));
        	s.add(sk);
        	sea[r][c] = sk;
        }
        while(king < C) {
        	move_shark();
        }
        sb.append(result);
        System.out.println(sb.toString());
        
    }
    static void move_shark() {
    	
    	// 낚시왕의 이동 & 상어 사라짐
    	for(int i=0; i<R; i++) {
    		if(sea[i][king] != null) {
    			Shark sk = sea[i][king];
    			result += sk.z;
    			s.remove(sk);
    			sea[i][king] = null;
    			break;
    		}
    	}
    	// 상어의 이동
    	int nx, ny;
    	int size = s.size();
    	for(int i = size-1; i>=0; i--) {
    		Shark sk = s.get(i);
    		if(sea[sk.r][sk.c] == sk)
    			sea[sk.r][sk.c] = null;
    		for(int k=0; k<sk.s; k++) {
    			nx = sk.r + dx[sk.d];
    			ny = sk.c + dy[sk.d];
    			if(nx < 0 || ny < 0 || nx >= R || ny >= C) {
    				if(sk.d % 2 == 0) sk.d -= 1;
    				else sk.d += 1;
    				nx = sk.r + dx[sk.d];
        			ny = sk.c + dy[sk.d];
    			}
    			sk.r = nx; sk.c = ny;
    		}
    		if(sea[sk.r][sk.c] != null && s.indexOf(sea[sk.r][sk.c]) > i) {
    			if(sea[sk.r][sk.c].z > sk.z) {
    				s.remove(i);
    			}
    			else {s.remove(sea[sk.r][sk.c]); sea[sk.r][sk.c] = sk;}
    		}
    		else {sea[sk.r][sk.c] = sk;}
    		
    	}
    	king ++;
    }
}

한 줄 씩 살펴보자!

class Shark{
	public 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;
	}
}

상어 클래스를 선언한다.

행, 열의 위치와 속력, 이동 방향, 크기 정보를 받아서 객체에 저장할 것이다.

static int R;
static int C;
static int M;
static int result;
static List<Shark> s;
static int king;
static Shark[][] sea;
static int[] dx = {0, -1, 1, 0, 0};
static int[] dy = {0, 0, 0, 1, -1};

입력받은 배열의 크기를 R, C에 저장한다. 상어의 수는 M에 저장해준다. result에 잡은 상어 크기의 합을 저장할 것이다. s를 통해 상어 객체 리스트를 만들어준다.

낙시왕의 위치 열은 king 변수에 저장해준다.

sea 배열을 통해 각 위치에 있는 상어를 저장해준다.

상어는 총 4가지의 이동방향을 갖는데 이때 1~4로 입력이 들어오기 때문에 배열을 1~4로 바로 사용할 수 있게 순서와 인덱스를 맞추어주었다.

R = Integer.parseInt(st.nextToken());
C = Integer.parseInt(st.nextToken());
sea = new Shark[R][C];

int T = Integer.parseInt(st.nextToken());
s = new ArrayList<>();
for(int i=0; i<T; i++) {
	st = new StringTokenizer(bf.readLine(), " ");
	int r = Integer.parseInt(st.nextToken()) -1;
	int c = Integer.parseInt(st.nextToken()) -1;
	Shark sk = new Shark(r, c, Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()));
	s.add(sk);
	sea[r][c] = sk;
}
while(king < C) {
	move_shark();
}
sb.append(result);
System.out.println(sb.toString());

R, C에 값을 입력 받는다. sea[r][c] 위치에 상어가 존재하면 그 객체를 저장한다.

낙시왕은 열 위를 이동하며 움직이고 격자를 벗어나면 더 이상 낚시를 할 수 없기 때문에 낙시왕이 C보다 작을 때까지만 낚시를 진행한다.

static void move_shark() {
	// 낚시왕의 이동 & 상어 사라짐
	for(int i=0; i<R; i++) {
		if(sea[i][king] != null) {
			Shark sk = sea[i][king];
			result += sk.z;
			s.remove(sk);
			sea[i][king] = null;
			break;
		}
	}

낙시왕은 현재 위치한 열의 가장 가까운 행(즉, 0에 가장 가까운 행을 가지는 곳)의 상어를 잡는다. 행을 증가시키며 그 위치에 상어가 존재하면(null이 아니면) 상어를 이차원 배열과 list에서 제거하고 상어의 크기를 더해준다.

제거하면서 초기화를 해주는 것을 잊지 말자

int nx, ny;
int size = s.size();
for(int i = size-1; i>=0; i--) {
	Shark sk = s.get(i);
	if(sea[sk.r][sk.c] == sk)
		sea[sk.r][sk.c] = null;
	for(int k=0; k<sk.s; k++) {
		nx = sk.r + dx[sk.d];
		ny = sk.c + dy[sk.d];
		if(nx < 0 || ny < 0 || nx >= R || ny >= C) {
			if(sk.d % 2 == 0) sk.d -= 1;
			else sk.d += 1;
			nx = sk.r + dx[sk.d];
  			ny = sk.c + dy[sk.d];
		}
		sk.r = nx; sk.c = ny;
	}

상어를 삭제하는 과정이 있기 때문에 인덱스를 뒤에서부터 조사해준다. 또한 list의 크기가 가변적이기 때문에 변수에 값을 저장하여 대체한다.

상어 객체를 받아 이차원 배열에서 상어 객체를 제거하고 다음 이동 위치를 조사하여 넘어가는 경우 방향을 바꿔가면서 다음 위치를 지정한다.

if(sea[sk.r][sk.c] != null && s.indexOf(sea[sk.r][sk.c]) > i) {
  			if(sea[sk.r][sk.c].z > sk.z) {
  				s.remove(i);
  			}
  			else {s.remove(sea[sk.r][sk.c]); sea[sk.r][sk.c] = sk;}
  		}
  		else {sea[sk.r][sk.c] = sk;}
  		
  	}
  	king ++;
  }
}

이동할 위치가 결정된 경우 그 위치에 다른 상어가 존재하면 & 그리고 그 상어가 이번 시간에 이동한 것이라면? (인덱스를 거꾸로 이동하고 있기 때문에 더 큰 인덱스를 가진 상어는 먼저 이동하는 것으로 체크 가능하다.)

그런 경우 크기가 작은 상어를 제거한다.

그 뒤 낙시왕의 위치를 이동해준다.