문제

과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 1점부터 k점까지의 점수로 분류하며, k점이 최상품의 사과이고 1점이 최하품의 사과입니다. 사과 한 상자의 가격은 다음과 같이 결정됩니다.

한 상자에 사과를 m개씩 담아 포장합니다.
상자에 담긴 사과 중 가장 낮은 점수가 p (1 ≤ p ≤ k)점인 경우, 사과 한 상자의 가격은 p * m 입니다.
과일 장수가 가능한 많은 사과를 팔았을 때, 얻을 수 있는 최대 이익을 계산하고자 합니다.(사과는 상자 단위로만 판매하며, 남는 사과는 버립니다)

예를 들어, k = 3, m = 4, 사과 7개의 점수가 [1, 2, 3, 1, 2, 3, 1]이라면, 다음과 같이 [2, 3, 2, 3]으로 구성된 사과 상자 1개를 만들어 판매하여 최대 이익을 얻을 수 있습니다.

(최저 사과 점수) x (한 상자에 담긴 사과 개수) x (상자의 개수) = 2 x 4 x 1 = 8
사과의 최대 점수 k, 한 상자에 들어가는 사과의 수 m, 사과들의 점수 score가 주어졌을 때, 과일 장수가 얻을 수 있는 최대 이익을 return하는 solution 함수를 완성해주세요.

아이디어

1. 우선 score 배열을 Array.prototype.sort() 메소드를 통해 내림차순으로 정렬한다. (정수형으로 내림차순 정렬이기 때문에 꼭 sort 메소드에 compareFunction을 콜백으로 넘겨주어야 한다!)
2. 한 박스에 들어가는 사과의 개수가 m개이고, 이미 내림차순으로 정렬되어 있으므로, m번째 요소들이 각 박스에 들어가는 최소 점수의 사과이다. 따라서, 해당 사과의 점수에 m을 곱한 값을 answer에 더해준다.

틀린 코드 (시간 초과)

코드

function solution(k, m, score) {
    var answer = 0;
    var lowest = 0;
    
    score = score.sort((a, b) => b-a);
    
    //이익이 발생하지 않는 경우 제외
    while(m <= score.length) {
        lowest = score[m-1];
        answer += lowest * m;
        score = score.slice(m);
    }
    
    return answer;
}

시간 초과 원인

• score 배열 안에서 일관된 간단한 인덱스로 접근하려고 일부러 Array.prototype.slice() 메소드를 사용했는데 이것이 시간 초과를 일으킨 것 같다. 매 번(while을 돌 때마다) 인덱스 m부터 끝까지 새로운 배열을 만들어내야 해서 생각보다 시간이 많이 걸리는 방법이었던 것이다.

통과한 코드

개선 사항

• Array.prototype.slice() 메소드를 쓰지 않고 반복문 하나 안에서 별다른 로직 처리 없이 인덱스만으로 처리하도록 만들었다.

코드

function solution(k, m, score) {
    var answer = 0;
    var lowest = 0;
    
    score = score.sort((a, b) => b-a);
    
    for(let i=0; i<score.length; i+=m) {
        //이익이 발생하지 않는 경우 제외
        if(m > score.length)
            break;
        //score.length를 인덱스가 넘어선 경우
        if(i >= score.length || i+m-1 >= score.length)
            break;
        lowest = score[i+m-1];
        answer += lowest * m;
    }
    
    return answer;
}

다른 좋은 풀이

나는 score를 내림차순으로 정렬해서 나중에 인덱스가 넘치는 경우를 if문으로 따로 걸러주어야 했는데, 아래와 같은 코드로 진행하면 그러는 경우를 미리 방지하고 깔끔하게 m의 배수의 길이를 가진 score를 가지고 시작할 수 있다.

• 우선, 오름차순으로 정렬한다.
• 나머지 사과는 이익 창출에 도움이 안되므로, 해당 부분을 제외해버리고 새로운 배열로 시작!

const sortedScore = score.sort((a, b) => a - b)).slice(score.length % m);