당구 연습

2025-04-10

#프로그래머스#JAVA

문제 링크

문제 풀이 시간 : 30분

문제 풀이 :

처음에는 어떻게 풀어야지? 라는 생각이 계속 들었는데

문제를 읽다보니 입사각과 반사각이 같다는 걸 보고 그냥 대칭이동 시키면 되지 않을까? 라는 생각으로 접근했다.

예를 들어, 아래와 같은 상황이 있다면

여기서 흰공이 검은공을 맞출 수 있는 경우는 아래와 같이 4가지 경우이다.

하지만 여기서 파란 선의 경로는 벽에 튕기기 전 검은공에 닿아버리므로 제외하고, 나머지 3가지 경로에 대해 최소 경로를 구하면 된다.

이 상황에서 우리는 검은색 공을 아래 그림처럼 각 벽에 대해 대칭이동 시킨다.

그리고 흰공과 대칭이동 된 검은 공들에 대해 거리를 구하면 간단하게 구할 수 있다.

나는 이걸 구현하기 위해서 각 공에 대해 각 벽과 대칭이동 시킨 위치를 구했다.

Java

int[][] candi = {{tarX,2*n-tarY},{tarX,-tarY},{-tarX,tarY},{2*m-tarX,tarY}}; //위쪽 벽, 아래쪽 벽, 왼쪽 벽, 오른쪽 벽

그 후, 위 사진에서의 파란색 경로와 같이 벽에 부딪히기 전 검은 공에 부딪히는지를 검사해줬다.

Java

if ((j == 0 && startX == tarX && startY < tarY) || //위쪽
    (j == 1 && startX == tarX && startY > tarY) || //아래쪽
    (j == 2 && startY == tarY && startX > tarX) || //왼쪽
    (j == 3 && startY == tarY && startX < tarX)) { //오른쪽
    continue;
}

그리고 가능한 경로에 대해 거리를 계산해주면 된다

Java

int dir_x = Math.abs(startX- candi[j][0]);
int dir_y = Math.abs(startY - candi[j][1]);
int temp = dir_x * dir_x + dir_y * dir_y;
                
min = Math.min(min, temp);

전체 코드

Java

import java.util.*;

class Solution {
    
    public int[] solution(int m, int n, int startX, int startY, int[][] balls) {
        int[] answer = new int[balls.length];
        
        for(int i=0;i<balls.length;i++){
            int tarX = balls[i][0];
            int tarY = balls[i][1];
            int min = Integer.MAX_VALUE;
            
            int[][] candi = {{tarX,2*n-tarY},{tarX,-tarY},{-tarX,tarY},{2*m-tarX,tarY}};
            
            for(int j=0;j<4;j++){
                if ((j == 0 && startX == tarX && startY < tarY) ||
                    (j == 1 && startX == tarX && startY > tarY) ||
                    (j == 2 && startY == tarY && startX > tarX) ||
                    (j == 3 && startY == tarY && startX < tarX)) {
                    continue;
                }
                
                int dir_x = Math.abs(startX- candi[j][0]);
                int dir_y = Math.abs(startY - candi[j][1]);
                int temp = dir_x * dir_x + dir_y * dir_y;
                
                min = Math.min(min, temp);
            }
            answer[i]=min;
        }
        
        return answer;
    }
}