最小的K个数

题目描述

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

思路一：堆排序，小根堆，每次找到取出最小的根

代码一：

    void Heap(vector<int> &arr,int len,int Root)
    {
        int index;
        while(1)
        {
            if(Root*2+2 < len )//有左,右孩子
             {
                 //左右孩子中大的
                // printf("999\n");
                index = arr[Root*1+1] < arr[Root*2+2] ? Root*1+1 : Root*2+2;
             }
             else if(Root*1+1 < len)//只有左孩子
            {
                index = Root*1+1;
            }
            else//没孩子
                break;
            if(arr[index] < arr[Root]) //与根交换
             {
                 arr[index] = arr[index]^arr[Root];
                 arr[Root] = arr[index]^arr[Root];
                 arr[index] = arr[index]^arr[Root];
                 Root = index;
             }
             else
                break;
        }
    } 
    vector<int> GetLeastNumbers_Solution(vector<int> &input, int k) {
        vector<int> vec;
        int len = input.size();
        if(k > input.size()||k==0)  return vec;
         for(int i = len/2-1;i >=0 ;i--)
            Heap(input,len,i);
        
         for(int i = len-1;i>=0;i--)
        {
             vec.push_back(input[0]);
             if(vec.size() == k)
                 break;
            //交换
            input[0] = input[0]^input[i];
            input[i] = input[0]^input[i];
            input[0] = input[0]^input[i];
            //调整根
            Heap(input,i,0);
        }
        return vec;
    }