[리트코드] 347. 상위 K 빈도 요소 (Python)

Top K Frequent Elements - LeetCode

문제

배열에서 가장 많이 출현한 원소를 k개 반환해라

 

과정

1. collections 모듈의 Counter 클래스를 사용해 리스트에서의 원소의 개수를 구한다.
2.  
3. for문을 이용해 리스트로 출력한다. 

 

입력

[1,1,1,2,2,3]

출력

2

 

풀이

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        ans = collections.Counter(nums).most_common(k)
        answer = []
        for i in (ans):
            answer.append(i[0])
        return answer

 

한 줄씩 코드 해석해보기

ans = collections.Counter(nums).most_common(k)

• Counter(nums).most_common(k)을 이용해 입력받은  nums 리스트 중 가장 많이 등장한 숫자들을 k개 저장한다.
• 예제를 입력하면 ans 에는 [(1,3), (2,2)] 값이 저장된다.

answer = []
for i in (ans):
    answer.append(i[0])
return answer

• 정답을 출력할 리스트를 만든 후 ans의 첫번째 값을 넣어준다.

 

한 줄 풀이

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
    	return [i[0] for i in Counter(nums).most_common(k)]