标签：面试题 int 个数 40 high num low input pivot

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

解题思路

• 排序后遍历（相当于简化后的暴力）O(logn)
• 借助快排的Partition思想O(n)

上代码（C++香）

法一：排序后遍历（相当于简化后的暴力）

class Solution {
public:
    void mySwap(vector<int> &num, int i, int j){
        int temp = num[j];
        num[j] = num[i];
        num[i] = temp;
    }

    int myPartition(vector<int> &num, int low, int high){
        int pivot = num[low];
        while(low < high){
            // 将右边比pivot小的放到左边
            while(low < high && num[high] >= pivot)
                high--;
            mySwap(num, low, high);
            while(low < high && num[low] <= pivot)
                low++;
            mySwap(num, low, high);
        }
        return low;
    }

    // 快排
    void QSort(vector<int> &num, int low, int high){
        if(low >= high)
            return ;
        int pivot = myPartition(num, low, high);
        QSort(num, low, pivot - 1);
        QSort(num, pivot + 1, high);
    }

    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ans;
        if(k > input.size())
            return ans;
        QSort(input, 0, input.size() - 1);
        for(int i = 0; i < k; i++)
            ans.push_back(input[i]);
        return ans;
    }
};

法二：借助快排的Partition思想

  得到pivot如果与k-1相等，那么数组左边的数已经就比num[k-1]小了，所以找到这个pivot就行。

void mySwap(vector<int> &num, int i, int j){
    int temp = num[j];
    num[j] = num[i];
    num[i] = temp;
}

int myPartition(vector<int> &num, int low, int high){
    int pivot = num[low];
    while(low < high){
        // 将右边比pivot小的放到左边
        while(low < high && num[high] >= pivot)
            high--;
        mySwap(num, low, high);
        while(low < high && num[low] <= pivot)
            low++;
        mySwap(num, low, high);
    }
    return low;
}

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
    vector<int> ans;
    int length = input.size();
    if(k > length)
        return ans;
    int low = 0;
    int high = length - 1;
    int pivot = myPartition(input, low, high);
    while(pivot != k - 1){
        // k-1在左边继续找
        if(pivot > k - 1){
            high = pivot - 1;
            pivot = myPartition(input, low, high);
        }
        else{
            low = pivot + 1;
            pivot = myPartition(input, low, high);
        }
    }

    for(int i = 0; i < k; i++)
        ans.push_back(input[i]);
    return ans;
}

标签：面试题,int,个数,40,high,num,low,input,pivot
来源： https://www.cnblogs.com/flyingrun/p/13525777.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。