[Programmers] 11주차 - 아이템 줍기

문제 설명

다음과 같은 다각형 모양 지형에서 캐릭터가 아이템을 줍기 위해 이동하려 합니다.

지형은 각 변이 x축, y축과 평행한 직사각형이 겹쳐진 형태로 표현하며, 캐릭터는 이 다각형의 둘레(굵은 선)를 따라서 이동합니다.

만약 직사각형을 겹친 후 다음과 같이 중앙에 빈 공간이 생기는 경우, 다각형의 가장 바깥쪽 테두리가 캐릭터의 이동 경로가 됩니다.

단, 서로 다른 두 직사각형의 x축 좌표 또는 y축 좌표가 같은 경우는 없습니다.

즉, 위 그림처럼 서로 다른 두 직사각형이 꼭짓점에서 만나거나, 변이 겹치는 경우 등은 없습니다.

다음 그림과 같이 지형이 2개 이상으로 분리된 경우도 없습니다.

한 직사각형이 다른 직사각형 안에 완전히 포함되는 경우 또한 없습니다.

지형을 나타내는 직사각형이 담긴 2차원 배열 rectangle, 초기 캐릭터의 위치 characterX, characterY, 아이템의 위치 itemX, itemY가 solution 함수의 매개변수로 주어질 때, 캐릭터가 아이템을 줍기 위해 이동해야 하는 가장 짧은 거리를 return 하도록 solution 함수를 완성해주세요.

제한사항

  • rectangle의 세로(행) 길이는 1 이상 4 이하입니다.
  • rectangle의 원소는 각 직사각형의 [좌측 하단 x, 좌측 하단 y, 우측 상단 x, 우측 상단 y] 좌표 형태입니다.
    • 직사각형을 나타내는 모든 좌표값은 1 이상 50 이하인 자연수입니다.
    • 서로 다른 두 직사각형의 x축 좌표, 혹은 y축 좌표가 같은 경우는 없습니다.
    • 문제에 주어진 조건에 맞는 직사각형만 입력으로 주어집니다.
  • charcterX, charcterY는 1 이상 50 이하인 자연수입니다.
    • 지형을 나타내는 다각형 테두리 위의 한 점이 주어집니다.
  • itemX, itemY는 1 이상 50 이하인 자연수입니다.
    • 지형을 나타내는 다각형 테두리 위의 한 점이 주어집니다.
  • 캐릭터와 아이템의 처음 위치가 같은 경우는 없습니다.

소스 코드

import java.util.*;
class Pair {
    int x;
    int y;
    public Pair(int x, int y) {
        this.x = x;
        this.y = y;
    }
}

class Solution {
    public int[] dx = {1, -1, 0, 0};
    public int[] dy = {0, 0, 1, -1};
    public int solution(int[][] rectangle, int characterX, int characterY, int itemX, int itemY) {
        int answer = 0;
        int[][] map = new int[101][101];
        int[][] visited = new int[101][101];
        int[][] d = new int[101][101];
        
        // 바깥 테두리 그리기(2배)
        for(int rec[] : rectangle) {
            int x1 = rec[0]*2, y1 = rec[1]*2;
            int x2 = rec[2]*2, y2 = rec[3]*2;
            for(int i=x1;i<=x2;i++) {
                for(int j=y1;j<=y2;j++) {
                    // 이미 선이 그려져있으면(map[i][j]가 2라면) 바깥 테두리가 아님
                    if((i==x1 || i==x2 || j==y1 || j==y2) && map[i][j] != 2) {
                        map[i][j] = 1;
                    }
                    else {
                        map[i][j] = 2;
                    }
                }
            }
        }
        
        // BFS
        Queue<Pair> q = new LinkedList<>();
        q.add(new Pair(characterX*2,characterY*2)); // 2배니까 x2
        visited[characterX*2][characterY*2] = 1;
        while(!q.isEmpty()) {
            Pair start = q.poll();
            for(int k=0;k<4;k++) {
                int nx = start.x + dx[k];
                int ny = start.y + dy[k];
                if(nx<0 || ny<0 || nx>100 || ny>100) continue; 
                if(map[nx][ny]==1 && visited[nx][ny]==0){
                    d[nx][ny] = d[start.x][start.y] + 1;
                    visited[nx][ny] =1;
                    q.add(new Pair(nx,ny));
                }
            }
        }
        
        answer = d[itemX*2][itemY*2]/2; // 2배니까 거리는 /2       
        return answer;
    }
}

Review

원래 크기로 바깥 테두리를 그렸을 때, 겹치는 부분을 피해가는건 갈 수 있는 방향의 count를 세는 방식으로 해결했지만 중앙에 빈 공간이 생기는 경우(테스트 2)를 해결하기는 어려웠다.
고민하다 질문하기에서 다각형을 2배로 그리면 된다는 힌트를 얻어서 수정했더니 푸는데 성공했다. 2배로 그린 후 BFS를 할 때에도 조건들을 2배로 수정해주어야 한다!!
풀어서 후련하다..😵

좋은 웹페이지 즐겨찾기