堆排序

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param input int整型vector 
     * @param k int整型 
     * @return int整型vector
     */
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        // write code here
        vector<int> result;
        for(int i = 0;i<k;i++){
            int size = input.size();
            HeapSort(input);
            result.push_back(input[0]);
            int last = input[size-1];
            input[0] = last;
            input.pop_back();
        }
        return result;
    }
    void HeapSort(vector<int>& input){
        int N = input.size();
        for(int i = N / 2;i >= 0;i--){
            doIt(input,i,N);
        }
    }
    void doIt(vector<int>& input,int k,int N){
        int value = input[k];
    //从0开始估i = 2 * k + 1
for(int i = 2 * k +1;i < N;i = 2 * i +1){ if(i < N-1 && input[i] < input[i+1]) i++; if(value >= input[i]) break; input[k] = input[i]; k = i; } input[k] = value; }

 

posted @ 2025-03-05 19:25  yizhixiaozhu  阅读(4)  评论(0)    收藏  举报