剑指offer 40.最小的 K 个数 python代码

电玩女神 2022-05-25 10:53 145阅读 0赞

# 题目 #

寻找数组中的最小的k个数，也叫topk问题。  
[牛客网测试地址][Link 1]  
**注意：**  
牛客网的提交需要将最终的结果排序

# 思路 #

快速排序的 partition() 方法，会返回一个整数 j 使得 a\[l…j-1\] 小于等于 a\[j\]，且 a\[j+1…h\] 大于等于 a\[j\]，此时 a\[j\] 就是数组的第 j 大元素。可以利用这个特性找出数组的第 K 个元素，这种找第 K 个元素的算法称为快速选择算法。

# 代码 #

# -*- coding: utf-8 -*-
    
    # !/usr/bin/env python
    
    # Time: 2018/5/13 15:39
    
    # Author: sty
    
    # File: 40. GetLeastNumbers.py
    
    class Solution:
        """
        最小的 K 个数
        practice:https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?tpId=13&tqId=11182&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
        """
        def find_kth_smallest(self, nums, k):
            l, h = 0, len(nums) - 1
            while l < h:
                j = self.partition(nums, l, h)
                if j == k:
                    break
                if j > k:
                    h = j - 1
                else:
                    l = j + 1
            return nums[k]
    
        def partition(self, nums, low, high):
            parti = nums[low]
            while low < high:
                while low < high and nums[high] >= parti:
                    high -= 1
                nums[low] = nums[high]
                while low < high and nums[low] <= parti:
                    low += 1
                nums[high] = nums[low]
            nums[low] = parti
            return low
    
        def GetLeastNumbers_Solution(self, tinput, k):
            res = []
            if k > len(tinput) or k <= 0:
                return res
            self.find_kth_smallest(tinput, k - 1)
            for i in range(k):
                res.append(tinput[i])
            return sorted(res)
    
    # a = Solution()
    # print(a.GetLeastNumbers_Solution([4,5,1,6,2,7,3,8], 4))

[Link 1]: https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?tpId=13&tqId=11182&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking