[백준 / BOJ 20056] 마법사 상어와 파이어볼

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

20056번: 마법사 상어와 파이어볼

 

문제

---

어른 상어가 마법사가 되었고, 파이어볼을 배웠다.

마법사 상어가 크기가 N×N인 격자에 파이어볼 M개를 발사했다. 가장 처음에 파이어볼은 각자 위치에서 이동을 대기하고 있다. i번 파이어볼의 위치는 (ri, ci), 질량은 mi이고, 방향은 di, 속력은 si이다. 위치 (r, c)는 r행 c열을 의미한다.

격자의 행과 열은 1번부터 N번까지 번호가 매겨져 있고, 1번 행은 N번과 연결되어 있고, 1번 열은 N번 열과 연결되어 있다.

파이어볼의 방향은 어떤 칸과 인접한 8개의 칸의 방향을 의미하며, 정수로는 다음과 같다.

7	0	1
6		2
5	4	3

마법사 상어가 모든 파이어볼에게 이동을 명령하면 다음이 일들이 일어난다.

1. 모든 파이어볼이 자신의 방향 di로 속력 si칸 만큼 이동한다.
   • 이동하는 중에는 같은 칸에 여러 개의 파이어볼이 있을 수도 있다.
2. 이동이 모두 끝난 뒤, 2개 이상의 파이어볼이 있는 칸에서는 다음과 같은 일이 일어난다.
   1. 같은 칸에 있는 파이어볼은 모두 하나로 합쳐진다.
   2. 파이어볼은 4개의 파이어볼로 나누어진다.
   3. 나누어진 파이어볼의 질량, 속력, 방향은 다음과 같다.
      1. 질량은 ⌊(합쳐진 파이어볼 질량의 합)/5⌋이다.
      2. 속력은 ⌊(합쳐진 파이어볼 속력의 합)/(합쳐진 파이어볼의 개수)⌋이다.
      3. 합쳐지는 파이어볼의 방향이 모두 홀수이거나 모두 짝수이면, 방향은 0, 2, 4, 6이 되고, 그렇지 않으면 1, 3, 5, 7이 된다.
   4. 질량이 0인 파이어볼은 소멸되어 없어진다.

마법사 상어가 이동을 K번 명령한 후, 남아있는 파이어볼 질량의 합을 구해보자.

 

입력

---

첫째 줄에 N, M, K가 주어진다.

둘째 줄부터 M개의 줄에 파이어볼의 정보가 한 줄에 하나씩 주어진다. 파이어볼의 정보는 다섯 정수 ri, ci, mi, si, di로 이루어져 있다.

서로 다른 두 파이어볼의 위치가 같은 경우는 입력으로 주어지지 않는다.

 

출력

---

마법사 상어가 이동을 K번 명령한 후, 남아있는 파이어볼 질량의 합을 출력한다.

 

풀이

---

 

시뮬레이션 & 구현 문제이다. (가장.... 못하게...ㅆ.....🤔)

각 파이어볼의 정보(r, c, m, s, d) 를 저장할 객체 class fireBall을 생성했다. 그리고 N x N 격자를 ArrayList 2차원 배열로 구현했다. (List<fireBall>[][] map) 

map[i][j] 에는 (i, j)에 위치한 파이어볼 객체가 저장되어있다. 그리고 모든 파이어볼 저장에는 List<fireBall> fireBallList을 이용했다. 

 

1. 파이어볼 이동

fireBallList 에 저장된 파이어볼들을 이동한다. 이 때, 속도를 % N 해주는 이유는 속도가 최대 1,000 으로 N보다 클 수 있기 때문에 무의미한 반복을 방지하기 위함이다. 그렇게 이동한 파이어볼들을 map에 저장해준다. 

2. 파이어볼 합체 & 분열

map을 순회하면서 size가 2 미만이면 clear해준다. 2 이상이면 모든 질량과 속도를 합한 값을 구하면서, 방향이 모두 짝수 (또는 홀수)로 일치하는지 확인한다. 합친 파이어볼은 fireBallList에서 삭제해주고, 새로 분열한 4개의 파이어볼을 생성해 fireBallList에 저장해준다.

마지막은 fireBallList에 저장된 파이어볼들의 질량을 모두 더해서 출력해준다!!

 

순서대로하면 어렵지 않은 것같은데.... 나에겐 아직 구현문제는 머리가 아프다....흑🤮

 

코드

---

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

public class Main_BOJ_20056_마법사상어와파이어볼 {
    static class fireBall{
        int r;  // row
        int c;  // col
        int m;  // mass
        int s;  // speed
        int d;  // direction

        fireBall(int r, int c, int m, int s, int d){
            this.r = r;
            this.c = c;
            this.m = m;
            this.s = s;
            this.d = d;
        }
    }
    static List<fireBall>[][] map;
    static List<fireBall> fireBallList = new ArrayList<>();
    static int dr[] = {-1,-1,0,1,1,1,0,-1};
    static int dc[] = {0,1,1,1,0,-1,-1,-1};
    
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));  
        StringTokenizer stringTokenizer = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(stringTokenizer.nextToken());
        int M = Integer.parseInt(stringTokenizer.nextToken());
        int K = Integer.parseInt(stringTokenizer.nextToken());

        map = new List[N][N];
        for(int i = 0 ; i < N ;i++){
            for(int j = 0 ; j < N; j++){
                map[i][j] = new ArrayList<>();
            }
        }

        for(int i = 0; i < M ;i++){
            stringTokenizer = new StringTokenizer(br.readLine());
            int r = Integer.parseInt(stringTokenizer.nextToken());
            int c = Integer.parseInt(stringTokenizer.nextToken());
            int m = Integer.parseInt(stringTokenizer.nextToken());
            int s = Integer.parseInt(stringTokenizer.nextToken());
            int d = Integer.parseInt(stringTokenizer.nextToken());

            fireBallList.add(new fireBall(r-1, c-1, m, s, d));
        }

        while(K-- > 0){
            // 이동
            for(fireBall fBall : fireBallList){
                // s 가 최대 1,000으로 N을 초과할 수 있기 때문에 % N
                int nr = fBall.r + dr[fBall.d] *(fBall.s % N);
                int nc = fBall.c + dc[fBall.d] *(fBall.s % N);

                if(nr >= N)
                    nr -= N;
                else if(nr < 0)
                    nr += N;
                if(nc >= N)
                    nc -= N;
                else if(nc < 0)
                    nc += N;

                fBall.r = nr;
                fBall.c = nc;

                map[fBall.r][fBall.c].add(fBall);

            }

            // 파이어볼이 두개 이상인곳 확인
            for(int i = 0 ; i < N; i++){
                for(int j = 0 ; j < N; j++){
                    // 2개 미만이면 지우고 continue;
                    if(map[i][j].size() < 2){
                        map[i][j].clear();
                        continue;
                    }
                    
                    // 2개 이상이면
                    // 1. 방향이 모두 짝수(홀수)인지 아닌지 확인
                    boolean even = map[i][j].get(0).d % 2 == 0 ? true : false;
                    boolean odd = map[i][j].get(0).d % 2 == 1 ? true : false;

                    int mSum = 0;   // 질량 합 
                    int sSum = 0;   // 속도 합

                    for(fireBall f : map[i][j]){
                        mSum += f.m;
                        sSum += f.s;
                        // 방향이 짝수인지
                        even = even && f.d % 2 == 0 ? true : false;
                        // 방향이 홀수인지
                        odd = odd && f.d % 2 == 1 ? true : false;

                        fireBallList.remove(f);
                    }

                    int nMass = Math.floorDiv(mSum, 5);
                    int nSpeed = Math.floorDiv(sSum, map[i][j].size());
                    map[i][j].clear();

                    if(nMass == 0)
                        continue;

                    // 방향이 모두 홀수(짝수) 이면
                    if(even || odd){
                        // 방향이 0, 2, 4, 6
                        for(int k = 0 ; k < 7 ; k += 2)
                            fireBallList.add(new fireBall(i,j,nMass, nSpeed, k));
                    }
                    // 아니면
                    else{
                        // 방향이 1, 3, 5, 7
                        for(int k = 1; k < 8 ; k+=2)
                            fireBallList.add(new fireBall(i, j, nMass, nSpeed, k));
                    }
                }
            }
        }

        int ans = 0;
        for(fireBall f : fireBallList)
            ans += f.m;

        System.out.println(ans);
    }
}