[Java] 프로그래머스 문제 - 택배 상자 꺼내기

연달아서 풀어본 문제는 프로그래머스의 택배 상자 꺼내기 이라는 문제를 풀어보았다.

---

문제 설명

1 ~ n의 번호가 있는 택배 상자가 창고에 있습니다. 당신은 택배 상자들을 다음과 같이 정리했습니다.

왼쪽에서 오른쪽으로 가면서 1번 상자부터 번호 순서대로 택배 상자를 한 개씩 놓습니다. 가로로 택배 상자를 w개 놓았다면 이번에는 오른쪽에서 왼쪽으로 가면서 그 위층에 택배 상자를 한 개씩 놓습니다. 그 층에 상자를 w개 놓아 가장 왼쪽으로 돌아왔다면 또다시 왼쪽에서 오른쪽으로 가면서 그 위층에 상자를 놓습니다. 이러한 방식으로 n개의 택배 상자를 모두 놓을 때까지 한 층에 w개씩 상자를 쌓습니다.

• 위 그림은 w = 6일 때 택배 상자 22개를 쌓은 예시입니다.

다음 날 손님은 자신의 택배를 찾으러 창고에 왔습니다. 당신은 손님이 자신의 택배 상자 번호를 말하면 해당 택배 상자를 꺼내줍니다. 택배 상자 A를 꺼내려면 먼저 A 위에 있는 다른 모든 상자를 꺼내야 A를 꺼낼 수 있습니다. 예를 들어, 위 그림에서 8번 상자를 꺼내려면 먼저 20번, 17번 상자를 꺼내야 합니다.

당신은 꺼내려는 상자 번호가 주어졌을 때, 꺼내려는 상자를 포함해 총 몇 개의 택배 상자를 꺼내야 하는지 알고 싶습니다.

창고에 있는 택배 상자의 개수를 나타내는 정수 n, 가로로 놓는 상자의 개수를 나타내는 정수 w와 꺼내려는 택배 상자의 번호를 나타내는 정수 num이 매개변수로 주어집니다. 이때, 꺼내야 하는 상자의 총개수를 return 하도록 solution 함수를 완성해 주세요.

---

제한 사항

• 2 ≤ n ≤ 100
• 1 ≤ w ≤ 10
• 1 ≤ num ≤ n

---

입출력 예

n	w	num	result
22	6	8	3
13	3	6	4

---

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다. 꺼내야 하는 상자의 총 개수는 8번 상자를 포함해 3개입니다.

입출력 예 #2

6번 상자를 꺼내려면 먼저 13, 12, 7번 상자를 꺼내야 합니다.

따라서 4를 return 합니다.

---

풀이 과정

문제를 보자마자 아래와 같이 생각하여 풀자고 생각했다.

1. 박스 담을 공간 만들기(높이 , 너비를 계산하여 공간 만들기)
2. 박스 쌓기
   1. [왼→오] 로 쌓다가 [오→왼] 으로 변경하는 로직 및 새로운 행 시작을 위한 값 초기화
   2. 박스에 값 담기
   3. 최종적으로 꺼내고자 하는 박스 열 찾는 작업
   4. 다음 열 계산
3. 목표 박스를 꺼내기 위해 몇 번 열에 손을 대야 하는지 꺼내는 작업
   1. 해당 열의 값이 0일 경우 빈 박스라는 것을 의미하기에 체크 안함
   2. 해당 열의 값이 0이 아닌 경우 꺼내야 하는 박스기에 결과 값 증가
   3. 최종적으로 꺼내고자 하는 박스 번호와 일치하는 경우 반복문 빠져나오기

---

코드

import java.util.*;
class Solution {
    public int solution(int n, int w, int num) {
        int answer = 0;
        
        // 박스 담을 공간 만들기
        int h = n%w > 0 ? n/w + 1 : n/w;
        int[][] boxlist = new int[h][w];
        
        // 0: 왼→오, 1: 오→왼
        int direction = 0;
        int col = 0;
        int row = 0;
        
        // 목표 박스의 열 인덱스
        int targetCol = 0;
        
        // 박스 쌓기
        for( int i = 0; i < n; i++ )
        {
            if ( i > 0 && i % w == 0 )
            { // 새로운 행 시작
                row++;
                if (direction == 0) 
                {
                    direction = 1;
                    col = w - 1;
                }
                else
                {
                    direction = 0;
                    col = 0;
                }
            }
            boxlist[row][col] = (i + 1);
            
            // 목표 박스의 열 저장
            if (num == (i + 1)) targetCol = col;
            
            // 다음 열 계산
            if (direction == 0) col++;
            else col--;
        }
        
        // 목표 박스를 꺼내기 위해 몇 번 손을 대야 하는지 계산
        for (int i = boxlist.length - 1; i >= 0; i-- )
        {
            if (boxlist[i][targetCol] != 0) answer++;
            if (boxlist[i][targetCol] == num) break;
        }
        
        return answer;
    }
}