用堆实现最小的K个数的查找:输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    vector<int> heap;
    void insert(int val){
        heap.push_back(val);
        int index=heap.size()-1;
        int father=(index-1)/2;
        while(index>0){
            if(heap[index]>heap[father]){
                swap(heap[index],heap[father]);
                index=father;
                father=(index-1)/2;
            }
            else
                break;
        }
    }

    void pop(int k,int val){
        heap[0]=val;
        int max_index=k-1;
        int index=0;
        int lchild;
        int rchild;
        while(index<max_index){
            lchild=index*2+1;
            rchild=index*2+2;
            if(lchild>max_index)
                break;
            else if(rchild>max_index){
                if(heap[index]<heap[lchild]){
                    swap(heap[index],heap[lchild]);
                    index=lchild;
                }
                else
                    break;
            }
            else{
                int bigger=heap[lchild]>=heap[rchild]?lchild:rchild;
                if(heap[index]<heap[bigger]){
                    swap(heap[index],heap[bigger]);
                    index=bigger;
                }
                else
                    break;
            }
        }
    }

    void swap(int &val1,int &val2){
        int temp=val1;
        val1=val2;
        val2=temp;
    }

    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int size=input.size();
        for(int i=0;i<k;i++){  //先将数组中的前k个数存入大顶堆中,即初始化长度为k的大顶堆
            insert(input[i]);
        }
//剩下的从第k+1个数开始与大顶堆的顶进行比较,如果值比顶的值还大,说明这个数不可能是最小的前K个数,i++;
//如果值比顶的值小,则调用pop()函数:将大顶堆的顶部元素删除,将第k+1个元素放在顶部,之后再进行一些列的比较后,
//将该值放在合适的位置构造成大顶堆
        for(int i=k;i<size;i++){
            if(input[i]<heap[0]){
                pop(k,input[i]);
            }
        }
        return heap;
    }
};

int main(){
    Solution s;
    int a[8]={4,5,1,6,2,7,3,8};
    vector<int> vec(a,a+8);
    vector<int> res=s.GetLeastNumbers_Solution(vec,4);
    for(int i:res){
        cout<<i;
    }

}

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页