과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 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 함수를 완성해주세요.
score
배열을 Array.prototype.sort()
메소드를 통해 내림차순으로 정렬한다. (정수형으로 내림차순 정렬이기 때문에 꼭 sort
메소드에 compareFunction
을 콜백으로 넘겨주어야 한다!)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;
}
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);