347. Top K Frequent Elements

https://leetcode.com/problems/top-k-frequent-elements/

Top K Frequent Elements - LeetCode

Given an integer array nums and an integer k, return the k most frequent elements. You may return the answer in any order.

 

내가 푼 것 :

이번에도 딕셔너리 사용해서 key 에 요소를 넣고 반복될때마다 +1 을 해주었다.

그리고 sorted를 해주는데 value 기준으로 하는 방법이 파이썬에서는 지원해줄것 같은 느낌이 들어서

찾아보다가 이상한걸 찾아서 적요해봤다.

lambda x: x[1] 은 key 인자로 사용되는 람다 함수로 카-값 쌍에서 값을 추출해서 정렬의 기준으로 사용한다.

x[1] 은 튜플의 두 번째 요소인 value 를 나타낸다. 정렬의 키로 사용한다. 아래에서 더 자세히 설명한다

class Solution(object):
    def topKFrequent(self, nums, k):
        resultDic = {}
        for num in nums:
            if num in resultDic:
                resultDic[num]+=1
            else:
                resultDic[num] = 1
        sorted_itmes = sorted(resultDic.items(),key=lambda x:x[1],reverse=True)
        topK = sorted_itmes[:k]
        return [item[0] for item in topK]

 

 

lambda 사용 예제를 몇개 만들어보면

익명 함수를 생성하는데 사용되는 키워드인데. 구조와 예제

lambda arguments: expression

1. 간단한 연산

add = lambda x, y : x+y
result = add(3,5)

결과 : 8

2. 정렬의 키로 사용

data = [(1,5), (3,2), (2,8)]
sorted(data, key=lambda x: x[1]) 

결과 : [(3,2),(1,5),(2,8)]

3. 맵 함수와 함께 사용

numbers = [1,2,3,4,5]
list(map(lambda x:x **2,numbers))

결과 : [1,4,9,16,25]

4. 필터 함수와 함께 사용

numbers = [1,2,3,4,5,6,7,8,9]
list(filter(lambda x: x %2==0,numbers)

결과 : [2,4,6,8]