题解 | 寻找第K大

#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param a int整型一维数组 
# @param n int整型 
# @param K int整型 
# @return int整型
#
class Solution:
    def quick_sort(self, arr:List[int]):
        if len(arr) <=1:
            return arr
        else:
            pivot = arr[0]
            left = [x for x in arr[1:] if x> pivot]
            right = [x for x in arr[1:] if x <= pivot]
        return self.quick_sort(left) + [pivot] + self.quick_sort(right)
        
    def findKth(self , a: List[int], n: int, K: int) -> int:
        # write code here
        output = self.quick_sort(a)
        return output[K-1]

或者可以用.sort(reverse=True)或者sorted(a,reverse=True)

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务