[JAVA] 낚시왕_17143

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

17143번: 낚시왕

낚시왕 성공

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

512 MB

38152

11186

6356

25.961%

문제

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

예제 입력 2 복사

100 100 0

예제 출력 2 복사

0

예제 입력 3 복사

4 5 4
4 1 3 3 8
1 3 5 2 9
2 4 8 4 1
4 5 0 1 4

예제 출력 3 복사

22

예제 입력 4 복사

2 2 4
1 1 1 1 1
2 2 2 2 2
1 2 1 2 3
2 1 2 1 4

예제 출력 4 복사

4

출처

• 문제를 만든 사람: baekjoon
• 데이터를 추가한 사람: djm03178
• 문제의 오타를 찾은 사람: hellojdh

알고리즘 분류

• 구현

 

 

package baekjoon;

import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

public class Main
{
    private static Scanner s = new Scanner(System.in);
    
    
   public static void main(String[] args) {
       class Shark {
          // 코테이므로 getter, setter, 생성자 제외
          int r; // x위치
          int c; // y위치
          int s; // 속력
          int d; // 이동방향 (1위 2아 3오 4왼)
          int z; // 크기
       }
      
      int R = s.nextInt(); // 격자판 크기 (가로)
      int C = s.nextInt(); // 격자판 크기 (세로)
      int M = s.nextInt(); // 상어의 수
      
      int answer = 0; // 정답
      
      Map<Integer, Shark> sharkMap = new HashMap<>(); // 상어 목록
      
      int sharkNum = 0; // 다음 초에 잡을 상어 번호
      int time = 0; // 시간
      
      // board가 0으로 초기화되어 있기에 1부터 루프 시작
      for(int i=1; i<=M; i++) {
         Shark shark = new Shark();
         shark.r = s.nextInt()-1; // 0부터 시작하도록
         shark.c = s.nextInt()-1;
         shark.s = s.nextInt();
         shark.d = s.nextInt();
         shark.z = s.nextInt();
         
         sharkMap.put(i, shark);
         
         if(time == shark.c) {
            // 1초 후 해당 열에 있는 상어를 잡을 예정
            if(sharkNum == 0) {
               sharkNum = i;
            }
            else {
               // r이 낮으면 (땅에 가까우면) 교체
               if(sharkMap.get(sharkNum).r > sharkMap.get(i).r) {
                  sharkNum = i;
               }
            }
         }
      }
      
      // C만큼 낚시꾼이 이동
      // 1. 낚시꾼이 이동한다.
      for(; time < C ;time++) {
         // 더 이상 잡을 상어가 없으면 break;
         if(M<=0) break;
         
         // 2. 가장 가까운 상어를 잡는다.
         if(sharkNum != 0) {
            answer+=sharkMap.get(sharkNum).z;
            sharkMap.remove(sharkNum);
            M--;
         }
         
         
         // 3. 각 상어들이 이동한다.
         int[][] board = new int[R][C]; // 낚시판

         sharkNum = 0; // 다음에 잡을 상어 초기화
         
         List<Integer> removeList = new LinkedList<>(); // 제거된 상어 리스트 : while 안에서 remove해버리면 iterator에서 ConcurrentModificationException 발생
         
         Iterator<Integer> sharkList = sharkMap.keySet().iterator();
         while(sharkList.hasNext()) {
            int sharkN = sharkList.next();
            
            Shark shark = sharkMap.get(sharkN);
            
            int newR = shark.r; // x위치
            int newC = shark.c; // y위치
            
            int tmpS = shark.s; // 속력
            int tmpD = shark.d; // 방향
            
            int z = shark.z;
            
            //해당 상어가 이동할 위치 파악
            switch(shark.d) {
            case 1:
            case 2:
               //위, 아래
               while(true) {
            	  //System.out.println(sharkN + " : " + newC);
                  //내려갈 때
                  if(tmpD == 2) {
                     if(newR + tmpS > R-1) {
                        //범위를 넘어서면 끝으로 이동.
                        tmpS -= (R-1-newR); //남은 거리 감소
                        newR = R-1; //끝으로 이동.
                        tmpD = 1; // 위로 향하게 방향 전환
                     }
                     else {
                        //넘어서지 않으면 거기로 이동.
                        newR += tmpS;
                        break;
                     }
                  }
                  else {
                     if(tmpS > newR) {
                        tmpS -= newR;
                        newR = 0;
                        tmpD = 2;
                     }
                     else {
                        newR -= tmpS;
                        break;
                     }
                  }
               }
               break;
            case 3:
            case 4:
               //오른쪽, 왼쪽
               //속력이 R-1-r 미만이 될 때까지 감소
               while(true) {
                  //오른쪽으로 갈 때
                  if(tmpD == 3) {
                     if(newC + tmpS > C-1) {
                        //범위를 넘어서면 끝으로 이동.
                        tmpS -= (C-1-newC); //남은 거리 감소
                        newC = C-1; //끝으로 이동.
                        tmpD = 4; // 위로 향하게 방향 전환
                     }
                     else {
                        //넘어서지 않으면 거기로 이동.
                        newC += tmpS;
                        break;
                     }
                  }
                  else {
                     if(tmpS > newC) {
                        tmpS -= newC;
                        newC = 0;
                        tmpD = 3;
                     }
                     else {
                        newC -= tmpS;
                        break;
                     }
                  }
               }
               
               break;
            }
            
            Boolean isDeleted = false;
            
            if(board[newR][newC] != 0) {
                // 이미 다른 상어가 존재한다면 크기가 작은 놈 제거
                if(sharkMap.get(board[newR][newC]).z < shark.z) {
                    //만약 이로 인해 지워진 게 다음 삭제 대상이면 교체
                    if(sharkNum == board[newR][newC]) {
                 	   isDeleted=true;
                 	   sharkNum = sharkN;
                    }
                    
                   removeList.add(board[newR][newC]);
                   board[newR][newC] = sharkN;
                }
                else {
                   removeList.add(sharkN);
                }
                M--;
             }
            else {
            	//없으면 그냥 넣기
            	board[newR][newC] = sharkN;
            }
            
            //갱신
            shark.c=newC;
            shark.r=newR;
            shark.d=tmpD;
            
            sharkMap.put(sharkN, shark);
            
            //다음에 잡을 상어인지 확인
            //위 로직에서 지워졌으면 패스
            if(!isDeleted && newC == time+1) {
               if(sharkNum == 0) {
                  sharkNum = sharkN;
               }
               else {
                  if(sharkMap.get(sharkNum).r > sharkMap.get(sharkN).r) {
                        sharkNum = sharkN;
                  }
               }
            }
                        
         }
         
         for(int num : removeList) {
        	 sharkMap.remove(num);
         }
      }

      System.out.println(answer);
      
   }
   
}