剑指 Offer 40. 最小的k个数

剑指 Offer 40. 最小的k个数

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

示例 1：

输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：

输入：arr = [0,1,2,1], k = 1
输出：[0]

使用了堆排

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> ans;
        heapSort(arr, arr.size(),k);
        for (int i = k-1; i >=0; i--) {
            ans.push_back(arr[i]);
        }
        return ans;
    }

    // 递归方式构建堆(len是arr的长度，index是第一个非叶子节点的下标)
    void adjust(vector<int> &arr, int len, int index)
    {
        int left = 2 * index + 1; // index的左子节点
        int right = 2 * index + 2;// index的右子节点

        int maxIdx = index;
        if (left arr[maxIdx])     maxIdx = left;
        if (right arr[maxIdx])     maxIdx = right;

        if (maxIdx != index)
        {
            swap(arr[maxIdx], arr[index]);
            adjust(arr, len, maxIdx);
        }
    }
    // 堆排序
    void heapSort(vector<int> &arr, int size,int k)
    {
        // 构建大根堆（从最后一个非叶子节点向上）
        for (int i = size / 2 - 1; i >= 0; i--)
        {
            adjust(arr, size, i);
        }

        // 调整大根堆
        for (int i = size-1; i >k-1; i--)
        {
            swap(arr[0], arr[i]);           // 将当前最大的放置到数组末尾
            adjust(arr, i, 0);              // 将未完成排序的部分继续进行堆排序
        }
    }
};

看题解，有大神使用快排的变形也可以实现