2021-01-16] Kth Largest Element in an Array

오늘의 문제:

leetcode.com/explore/challenge/card/january-leetcoding-challenge-2021/581/week-3-january-15th-january-21st/3606/

Explore - LeetCode

 

문제 예시를 보며 이해해보자.

<문제 예시 1-2>

문제에서 말했듯 주어진 배열에서 k 번째로 큰 값을 구하면 된다.

 

예시 1번을 정렬하면 [3,2,1,5,6,4] -> [1,2,3,4,5,6] 이고, k = 2니 6,5 -> 2번째 큰 값은 5 이다.

 

예시 2번을 보면, 중복되는 값들도 생기기 시작한다. 

 

[3,2,3,1,2,4,5,5,6] -> [1,2,2,3,3,4,5,5,6] 이고, k=4니 4번째로 큰 값을 찾는다.

 

중복값 또한 체크하는 것이라 6,5,5,4 -> 4번째로 큰 값은 4이다.

 

문제풀이 )

위에서 처럼 정렬된 리스트를 만들고 뒤에서 k번째의 값을 구한다.

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        return sorted(nums)[len(nums)-k] # 정렬된 List에서 뒤에서 k번째 값을 구한다.