[프로그래머스] 귤 고르기 - JavaScript

문제 출처

Lv.2 귤 고르기 - JavaScript

https://school.programmers.co.kr/learn/courses/30/lessons/138476

문제 설명

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

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

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

제한 조건

1 ≤ k ≤ tangerine의 길이 ≤ 100,000
1 ≤ tangerine의 원소 ≤ 10,000,000

입출력 예

k	tangerine	result
6	[1, 3, 2, 5, 4, 5, 2, 3]	3
4	[1, 3, 2, 5, 4, 5, 2, 3]	2
2	[1, 1, 1, 1, 2, 2, 2, 3]	1

답안

function solution(k, tangerine) {
    let result = 1
    // k의 값을 가진 변수로, 남아있는 귤의 개수를 추적하는 데 사용됩니다.
    let cnt = k
    let arr = []
    let duplicationCnt = {}
    for (const e of tangerine){
        duplicationCnt[e] = ( duplicationCnt[e] || 0 ) + 1;
    }
    
    //tangerine 배열의 각 원소(귤의 종류)를 순회하며 duplicationCnt 객체에 귤의 종류를 키로 하여 중복된 개수를 카운트합니다.
    for (const e of Object.entries(duplicationCnt)) {
      arr.push(e[1])
    }
    
    //duplicationCnt 객체의 각 엔트리(키와 값의 쌍)를 순회하며, 값만 arr 배열에 저장합니다. (즉, 각 귤 종류의 개수만 배열에 저장됩니다.)
    for (const e of arr.sort((a,b)=>b-a)) {
    // 정렬된 배열을 순회하며, 현재 원소(귤의 개수)가 cnt보다 작다면, cnt에서 해당 원소를 뺀 값을 다시 cnt에 저장하고 result 값을 1 증가시킵니다.
        if(e<cnt){
          cnt =  cnt - e
          result++
        } else { 
        return result
        }
    }
}

피드백

• 변수명을 보다 직관적이도록 작성하면 좋을 것 같습니다.
• sort를 for of 문 안에 작성하면 가독성이 저하되므로 밖으로 빼는 방법, toSorted메서드를 사용 하는 방법 등으로 개선하면 좋을 것 같습니다
• toSorted 의 경우 사본을 반환하기에 참조 관계를 명확히 알기 쉬운 장점도 존재합니다

정답 코드 수정

function solution(k tangerines) {
    let minimumTypes = 1;
    let remainingCount = k;
    let tangerineCounts = [];
    let tangerineFrequency = {};

    for (const tangerine of tangerines) {
        tangerineFrequency[tangerine] = (tangerineFrequency[tangerine] || 0) + 1;
    } 

    for (const [, count] of Object.entries(tangerineFrequency)) {
        tangerineCounts.push(count);
    }

    const sortedTangerineCounts = [...tangerineCounts].sort((a, b) => b - a);
    
    for (const count of sortedTangerineCounts) {
        if (count < remainingCount) {
            remainingCount -= count;
            minimumTypes++;
        } else { 
            return minimumTypes;
        }
    }
}