[Leetcode] 347. TOP K Frequent Elements

📝 문제

📎 347. TOP K Frequent Elements

상위 k 번 이상 등장하는 요소를 출력

Example

Input : nums = [1,1,1,2,2,3] , k = 2

• 1이 3번, 2는 2번 , 3은 1번 등장한다.
• k 는 2 이므로 빈도 수 순위로 2등까지 배열에 담아서 출력하면 된다.
• 즉 [1,2] 를 반환한다.

❌ 처음 작성 코드 - 오답

def solution(nums, k):
    ans = collections.defaultdict(int)
    result = []
    for char in nums:
        ans[char] += 1

    for char in ans:
        if ans[char] >= k:
            result.append(char)

    return result

풀이 의도는 다음과 같다.

• defaultdict 를 사용하여 ans 에 ([1:3, 2:2, 3:1]) 을 저장한다.
• 그리고 nums 을 돌면서 ans[1] = 3. ans[2] = 2, ans[3] = 1 을 저장한다.
• 마지막으로 ans 을 돌면서 ans 의 각 값이 k 이상이면 result 배열에 담는다.

틀렸다!!! 근데 왜?

문제를 잘 못 이해했다... 주어진 k, 즉 2 이상 등장하는 수를 배열에 담아 출력하면 되는건지 알았는데, 앞서 문제를 설명한 것과 같이 등장한 순서(k)를 기준으로 생각했어야 했다.

✅ 수정된 코드 - 정답

def solution(nums, k):
    ans = collections.Counter(nums)
    d = []
    result = ans.most_common(k)
    for idx, freq in result:
        d.append(idx)
    return d

• 파이썬의 Counter 를 사용하여 ans 에 빈도수를 저장한다.
• most_common : 가장 빈번하게 등장하는 k개의 요소와 그 빈도수를 튜플 리스트로 반환
• 그렇게되면, Counter({1:3, 2:2, 3:1}) 이 저장된다. 그리고 result 에는 상위 k번 이상 등장하는 [(1, 3), (2, 2)] 이 담긴다.
• for 문으로 result 에서 해당 원소(1,2) 를 반환할 배열에 담고 출력한다.

⭐️ most_common

    def most_common(self, n=None):
        '''List the n most common elements and their counts from the most
        common to the least.  If n is None, then list all element counts.

        >>> Counter('abracadabra').most_common(3)
        [('a', 5), ('b', 2), ('r', 2)]

        '''

위는 most_common 을 눌러 들어간 설명이다. 설명과 같이 most_common 에 매개변수만큼 등장한 값들을 저장한다.
.
.
코딩 테스트는 보통 idle 을 제공하지 않는다. 파이썬은 많은 라이브러리를 제공하지만 idle 이 없을 때도 편하고 자연스럽게 사용할 수 있도록 많이 연습하자.