PGMRS_택배 상자 꺼내기(JAVA)

아우 졸려 ...

https://school.programmers.co.kr/learn/courses/30/lessons/389478

프로그래머스

 

이번 문제는 배열 순회 / 규칙성 찾기 문제다. 

내가 유독 배열 순회 문제를 잘 헷갈리는 편이라서

천천히 문제를 정리하느라 주석이 무지하게 길다. 감안하고 볼 것

class Solution {
    // 발견하기
    // 1. '몇 개'의 상자를 꺼내야 하는가?를 묻는 문제이지 '몇 번'상자를 '어떤 순서로' 꺼내는 것인가?를 묻지 않았다.
    // 2. 그렇다면 왜 지그재그를 강조했을까? n개의 박스가 w로 '나누어 떨어지지 않을 때' 남는 박스가 어디에 위치해 있느냐에 따라 답이 달라지기 때문이다.
    // 3. 순회 방향에서의 규칙성 : n을 w로 나눈 몫이 홀수라면, n개의 박스가 모두 채워진 마지막 줄은 왼쪽 -> 오른쪽으로 순회한다. 짝수라면 그 반대이다.
    // 3-1. n을 w로 나눈 몫이 h, 나머지를 d라고 하자. h가 짝수이면, d는 h+1행의 0~d-1열까지를 차지한다. 반대로 홀수이면, d는 w-d~w-1열까지를 차지하게 된다.
    // 4. num번 상자를 꺼내야 한다면, num번 상자가 몇 행에 있는지 구해야 한다. 따라서, 3-1번 과정을 num에 대하여 한 번 더 수행한다. 3-1 과정의 'n'이 num이 된다.
    // 5. num / w를 한 값이 numH이고, 나머지가 numD라면, num번 상자가 몇 행(r) 몇 열(c)에 있는지 구할 수 있다. 3번에서 언급된 규칙을 바탕으로, 3-1번의 나머지 상자 배치 열에 c가 포함되는지 확인하여 h-r을 연산한 값이 답이 된다. 
    
    static class Idx {
        int r, c;
        Idx(int r, int c){
            this.r = r;
            this.c = c;
        }
        
        void printOut(){
            System.out.printf("[%d][%d]\n", r, c);
        }
    }
    
    static int h, d; // height of boxes (n/w, n%w)
    
    public int solution(int n, int w, int num) {
        h = n / w;
        d = n % w;
        
        Idx idx = where(num, w);
        
        // 네 가지 케이스가 나올 수 있다.
        // 1. h가 홀수이고 d가 0이 아닌 경우
        // 2. h가 홀수이고 d가 0인 경우
        // 3. h가 짝수이고 d가 0이 아닌 경우
        // 4. h가 짝수이고 d가 0인 경우
        // 이 중에서, d가 0이면 어떤 형태로든 행을 완성하게 되므로 무조건 답이 h - idx.r이다.
        int result = h - idx.r;
        if(d == 0) return result;
        
        // 이 스코프에서 걸러내는 경우는 idx.c가 나머지 상자 범위에 해당하는 열인 경우다.
        if(h % 2 == 0){
            // h가 짝수라면, d는 정방향 순회한다.
            if(idx.c <= d) return result+1;
        } else {
            // h가 홀수라면, d는 역방향 순회한다.
            if(idx.c >= w-d && idx.c < w-1) return result+1;
        }
        
        return result;
    }
    
    public Idx where(int num, int w){
        int r = num / w;
        int c = 0;
        if(r % 2 == 0){
            if(num % w == 0){
                r--;
                c = 0;
            } else {
                c = num % w - 1;
            }
        } else {
            if(num % w == 0){
                r--;
                c = w - 1;
            } else {
                c = w - (num % w);
            }
        }
        return new Idx(r, c);
    }
}

 

내가 뭔가 수학적 역량이 더 있었다면 ... 코드를 더 간추릴 수 있지 않았을까? 싶긴 하다.

특히 .. d != 0이고 h % 2 == 1인 경우의 나머지 상자 확인 조건이 좀. 찜찜.

정확성 테스트

테스트 1 〉	통과 (0.36ms, 80.8MB)
테스트 2 〉	통과 (0.28ms, 80.8MB)
테스트 3 〉	통과 (0.45ms, 89.9MB)
테스트 4 〉	통과 (0.26ms, 82.1MB)
테스트 5 〉	통과 (0.33ms, 81.9MB)
테스트 6 〉	통과 (0.27ms, 87.6MB)
테스트 7 〉	통과 (0.36ms, 72.3MB)
테스트 8 〉	통과 (0.50ms, 82.2MB)
테스트 9 〉	통과 (0.27ms, 89.9MB)
테스트 10 〉	통과 (0.33ms, 87.7MB)
테스트 11 〉	통과 (0.31ms, 74.8MB)
테스트 12 〉	통과 (0.42ms, 73.1MB)
테스트 13 〉	통과 (0.39ms, 84.1MB)
테스트 14 〉	통과 (0.27ms, 81.2MB)
테스트 15 〉	통과 (0.29ms, 75.8MB)
테스트 16 〉	통과 (0.30ms, 87MB)
테스트 17 〉	통과 (0.28ms, 79.3MB)
테스트 18 〉	통과 (0.34ms, 85.9MB)
테스트 19 〉	통과 (0.28ms, 73.8MB)
테스트 20 〉	통과 (0.39ms, 83.8MB)
테스트 21 〉	통과 (0.39ms, 77MB)
테스트 22 〉	통과 (0.30ms, 72.7MB)
테스트 23 〉	통과 (0.45ms, 79.1MB)
테스트 24 〉	통과 (0.31ms, 85.4MB)
테스트 25 〉	통과 (0.49ms, 89.4MB)

채점 결과

정확성: 100.0

합계: 100.0 / 100.0