[JS] 최소직사각형 - 프로그래머스

문제

명함 지갑을 만드는 회사에서 지갑의 크기를 정하려고 합니다. 다양한 모양과 크기의 명함들을 모두 수납할 수 있으면서, 작아서 들고 다니기 편한 지갑을 만들어야 합니다. 이러한 요건을 만족하는 지갑을 만들기 위해 디자인팀은 모든 명함의 가로 길이와 세로 길이를 조사했습니다.

아래 표는 4가지 명함의 가로 길이와 세로 길이를 나타냅니다.

명함 번호	가로 길이	세로 길이
1	60	50
2	30	70
3	60	30
4	80	40

가잘 긴 가로 길이와 세로 길이가 각각 80, 70 이기 때문에 80(가로) x 70(세로) 크기의 지갑을 만들면 모든 명함들을 수납할 수 있습니다. 하지만 2번 명함을 가로로 눕혀 수납한다면 80(가로) x 50(세로) 크기의 지갑으로 모든 명함들을 수납할 수 있습니다. 이때의 지갑 크기는 4000(=80x50) 입니다.

 

모든 명함의 가로 길이와 세로 길이를 나타내는 2차원 배열 sizes 가 매개변수로 주어집니다. 모든 명함을 수납할 수 있는 가장 작은 지갑을 만들 때, 지갑의 크기를 return 하도록 solution 함수를 완성해주세요.

 

제한사항

sizes의 길이는 1 이상 10,000 이하입니다.

• sizes의 원소는 [w, h] 형식입니다.
• w는 명함의 가로 길이를 나타냅니다.
• h는 명함의 세로 길이를 나타냅니다.
• w와 h는 1 이상 1,000 이하인 자연수입니다.

 

풀이

단순히 2차원 배열의 첫번째 요소 중 최대값 / 두번째 요소 중 최대값 을 구하는 문제였다면 간단했겠지만, 문제에서처럼 명함을 크기에 따라 눕힐 수 있다.

눕힐 수 있다는 말은 코딩 상으로 [w, h] 의 위치가 [h, w] 처럼 바뀔 수 있다는 말이 된다.

해서 먼저 2차원 배열 각각을 큰 값을 왼쪽, 작은 값을 오른쪽으로 스위칭하여 명함들 중 최대 길이가 다른 명함들을 포괄할 수 있도록 하였다.

    //큰 값을 왼쪽으로, 작은 값을 오른쪽으로 스위치
    for(let i = 0; i < sizes.length; i++)
    {
        if(sizes[i][0] < sizes[i][1])
        {
            let temp = sizes[i][1];
            sizes[i][1] = sizes[i][0];
            sizes[i][0] = temp;
        }
    }

 

이렇게만 되면 나머지는 간단하다. 각 인덱스를 돌면서 첫번째 요소의 최대값, 두번째 요소의 최대값을 구하고 그 값을 곱해주기만 하면 된다.

    let left_max = 0, right_max = 0;
    //양 사이드에서 최대값 구하기
    for(let i = 0; i<sizes.length; i++)
    {
        if(sizes[i][0] > left_max) left_max = sizes[i][0];
        if(sizes[i][1] > right_max) right_max = sizes[i][1];
    }
    answer = left_max * right_max;

 

완성된 코드는 다음과 같다.

function solution(sizes) {
    var answer = 0;
    //큰 값을 왼쪽으로, 작은 값을 오른쪽으로 스위치
    for(let i = 0; i < sizes.length; i++)
    {
        if(sizes[i][0] < sizes[i][1])
        {
            let temp = sizes[i][1];
            sizes[i][1] = sizes[i][0];
            sizes[i][0] = temp;
        }
    }
    
    let left_max = 0, right_max = 0;
    //양 사이드에서 최대값 구하기
    for(let i = 0; i<sizes.length; i++)
    {
        if(sizes[i][0] > left_max) left_max = sizes[i][0];
        if(sizes[i][1] > right_max) right_max = sizes[i][1];
    }
    answer = left_max * right_max;
    return answer;
}

 

실행

 

아쉬운 점

다만 이 코드를 풀면서 아쉬웠던 점으로, JS문법을 사용한 것이 아니라 거의 C++ 과 Vector 를 사용하는 기분으로 코드가 작성되었다.

해서 이 코드를 JS문법에 특화하여 리팩토링한 코드를 같이 첨부한다.

function solution(sizes) {
    // 각 인덱스마다 큰 값을 왼쪽, 작은 값을 오른쪽에 배치
    sizes = sizes.map(([w, h]) => w > h ? [w, h] : [h, w]);
    
    // 최대값 구하기
    const left_max = Math.max(...sizes.map(size => size[0]));
    const right_max = Math.max(...sizes.map(size => size[1]));

    return left_max * right_max;
}

 

값을 스위칭하는 부분은 map 과 삼항 연산자를 통하여 코드를 간결하게 작성하였다.

이후 최대값을 구하는 부분에서는 Math.max 를 통한 최대값 구하기, ... 스프레드 기법을 통한 각 인덱스를 전개하는 과정을 거친다.