과일 장수 - 프로그래머스

문제 설명
과일 장수가 사과 상자를 포장하고 있습니다. 사과는 상태에 따라 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 함수를 완성해주세요.

프로그래머스/과일 장수

 

입출력 예

k	m	score	result
3	4	[1, 2, 3, 1, 2, 3, 1]	8
4	3	[4, 1, 2, 2, 4, 4, 4, 4, 1, 2, 4, 2]	33

 

 

내 풀이

function computeMaxProfit(k, m, score) {
  const numBox = Math.floor(score.length / m);
  const availableScores = score.sort((a, b) => b - a).slice(0, numBox * m);

  const result = availableScores.reduce((acc, cur, index) => {
    if ((index + 1) % m === 0) {
      return (acc = acc + cur * m);
    }

    return acc;
  }, 0);

  return result;
}

 

 

고민을 해보니 결국 최대 이익을 내려면 내림차순으로 정렬을 해서 한 박스에 들어갈 사과의 개수가 정해지니 그것으로 몇 박스를 채울 수 있을지 알면 되었다.

 

우선 score.length를 한 박스에 들어갈 사과의 개수로 나눈 뒤 소수점을 버려 numBox 를 정의했다.

그리고 score 배열에서 내림차순으로 정렬한 뒤 필요한 수만 빼고 버려주는 작업을 했다.

 

이후 reduce를 사용했고, 각 박스에서 마지막에 위치할 인덱스 값의 조건에 위치한 그 박스의 최소값이라면 그 값에 한 상자에 담긴 사과 개수를 곱하여주고 더해줌으로써 결과 값을 도출해냈다.

 

다른 사람의 풀이

function solution(k, m, score) {
    let answer = 0;
    const sortedScore = score.slice().sort((a, b) => a - b).slice(score.length % m);
    for (let i = 0; i < sortedScore.length; i += m) {
        answer += sortedScore[i] * m;
    }
    return answer;
}

 

간결하고 깔끔했다.

우선 score.slice()로 원본 배열을 유지한 상태로 배열을 복사하였고 그것으로 오름차순 정렬을 한 뒤 나머지 값들을 slice 로 잘라내었다. 이후 반복문을 통해 상자의 최소값만 뽑아서 더해주기 위해 i를 m씩 증가시켰다.

 

 

내가 접근한 방식보다 두 가지 측면에서 더 좋다라는 생각이 들었다.

 

첫 번째로는 score에 바로 sort를 사용하면 함수의 파라미터로 전달된 배열은 원본 배열과 같은 메모리 주소를 참조하고 있기 때문에 원본 배열 자체가 변경되기 때문에 배열 복사를 위해 slice()를 사용한 점이었다. 

두 번째로는 reduce()는 내부적으로 콜백 함수를 호출해야 해서 콜백 함수 호출 비용이 추가되어 함수를 호출하는 오버헤드가 반복문보다 클 수 있어서 간단한 연산에 있어서는 for문이 더 빠른 경우가 많다.