题解 | 寻找第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)