[알고리즘] 프로그래머스 - 귤 고르기 (JS)

 

✅ 문제

경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 'k'개를 골라 상자 하나에 담아 판매하려고 합니다. 그런데 수확한 귤의 크기가 일정하지 않아 보기에 좋지 않다고 생각한 경화는 귤을 크기별로 분류했을 때 서로 다른 종류의 수를 최소화하고 싶습니다.

예를 들어, 경화가 수확한 귤 8개의 크기가 [1, 3, 2, 5, 4, 5, 2, 3] 이라고 합시다. 경화가 귤 6개를 판매하고 싶다면, 크기가 1, 4인 귤을 제외한 여섯 개의 귤을 상자에 담으면, 귤의 크기의 종류가 2, 3, 5로 총 3가지가 되며 이때가 서로 다른 종류가 최소일 때입니다.

경화가 한 상자에 담으려는 귤의 개수 k와 귤의 크기를 담은 배열 tangerine이 매개변수로 주어집니다. 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

 

✅ 풀이

우선 해시맵 자료구조를 사용해서 귤의 크기를 key, 개수를 value로 fruits에 담아주었다. (O(N))

다음 귤의 개수가 많은 순대로 귤의 크기를 내림차순하여 newTangerine에 담고, k가 0이 될 때까지 newTangerine을 선형탐색하면서 k에서 fruits[nowSize]를 빼나간다. k가 0일 때 answer 값이 곧 크기의 종류의 최솟값이다. 완전탐색으로 풀이하였다. (시간복잡도 O(N))

 

function solution(k ,tangerine) {
  let answer = 0;
  let i = 0;
  const fruits = {};
  
  // 한 상자에 담으려는 귤의 개수 k개를 고를 때 크기의 종류를 최소화 하자.
  for (const f of tangerine) {  
    if (fruits[f]) {
      fruits[f] += 1;
    }else {
      fruits[f] = 1;
    }
  }
  
  // 귤의 개수가 많은 순대로 크기를 내림차순 정렬
  const newTangerine = [...Object.keys(fruits).sort((a, b) => fruits[b] -fruits[a] )];
  
  // 상자에 모든 귤을 담을 때까지 반복
  while(k) {
    const nowSize = newTangerine[i];
    
    if (fruits[nowSize] - k >= 0) {
      answer += 1;
      k = 0;
      break;
    } else {
      answer += 1;
      k -= fruits[nowSize];
      i += 1;
    }
  }
  
  return answer;
}