과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 1점부터 k점까지의 점수로 분류하며, k점이 최상품의 사과이고 1점이 최하품의 사과입니다. 사과 한 상자의 가격은 다음과 같이 결정됩니다.
과일 장수가 가능한 많은 사과를 팔았을 때, 얻을 수 있는 최대 이익을 계산하고자 합니다.(사과는 상자 단위로만 판매하며, 남는 사과는 버립니다)
예를 들어, k = 3, m = 4, 사과 7개의 점수가 [1, 2, 3, 1, 2, 3, 1]이라면, 다음과 같이 [2, 3, 2, 3]으로 구성된 사과 상자 1개를 만들어 판매하여 최대 이익을 얻을 수 있습니다.
사과의 최대 점수 k, 한 상자에 들어가는 사과의 수 m, 사과들의 점수 score가 주어졌을 때, 과일 장수가 얻을 수 있는 최대 이익을 return하는 solution 함수를 완성해주세요.
처음 시도한 코드는 이랬다.
function solution(k, m, score) {
var answer = 0;
if(score.length < m) return 0
score.sort((x, y) => y - x)
while(score.length >= m) {
let box = score.splice(0, m)
let price = m * box[box.length - 1]
answer += price
}
return answer
}
이익이 발생할 수 없는(score 길이가 m보다 작은) 경우를 먼저 if 문으로 걸러내기
내림차순으로 정렬 후
while문을 돌며 박스에 담고 가격 책정 후 answer에 더하기
이렇게 했을 때 먼저 제공된 테스트케이스는 통과해서 채점을 해보니 테스트 24개 중에 테스트 11 ~ 15가 시간초과로 실패가 떴다.☹
아무래도 score의 길이 범위가 7 ≤ score의 길이 ≤ 1,000,000인데 while문을 사용한 것이 원인인 것 같아 다른 방법을 생각해보기로 했다.🤔🤔
score의 길이 만큼 다 돌지 않는 방법을 생각하다 score.length를 한 상자에 담을 개수(m)만큼 나눴다.
먼저 위와 똑같이 score를 내림차순으로 정렬해주고
score.length / m을 한 값을 box라는 변수를 만들어줬다.(소수점은 불필요하기 때문에 Math.floor()로 소수점 버리기)
그 다음 box값 만큼 for문을 돌며 값을 구하고 answer에 더해주면 끝!
function solution(k, m, score) {
var answer = 0;
score.sort((x, y) => y - x)
let box = Math.floor(score.length / m)
for(let i = 1; i <= box; i++) {
answer += score[(i * m) - 1] * m
}
return answer
}
✔문제출처: 프로그래머스