求最小的K个数

最新推荐文章于 2020-04-07 08:11:59 发布

原创最新推荐文章于 2020-04-07 08:11:59 发布 · 692 阅读

0 ·

CC 4.0 BY-SA版权

面试题专栏收录该内容

29 篇文章

订阅专栏

本文介绍了一种寻找数组中最小k个数的算法实现，通过递归划分的方法找到目标元素，避免了直接排序带来的高时间复杂度。文章提供了一个具体的C++实现案例，并强调了理解算法原理的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这样的题目写过很多，我这次要说明的是，一定要按照自己的想法来写，不要死记书上的解法，那样的话，你并没有转化为自己的东西

#include <stdio.h>
#include <stack>
#include <vector>
#include <iostream>
using namespace std;

vector<int> res;
void swap(vector<int> &v,int i,int j){
	int tmp=v[i];
	v[i]=v[j];
	v[j]=tmp;
}
void getmink(vector<int> v,int k,int b,int e){
	if(k==0||b>e)
		return;
	int i=b-1,split=v[e];
	for(int j=b;j<e;j++){
		if(v[j]<split){
			i++;
			swap(v,i,j);
		}
	}
	i++;
	swap(v,i,e);
	if(i-b+1<=k){
		for(int j=b;j<=i;j++)
		    res.push_back(v[j]);
		getmink(v,k-(i-b+1),i+1,e);
	}else{
		//一开始没有减一，导致死循环
		getmink(v,k,b,i-1);

	}
}

int main(){
	int a[]={1,0,8,7,6,5,33,2,1,5};
	vector<int> v(a,a+10);
	getmink(v,5,0,9);
	for(int i=0;i<res.size();i++)
		cout<<res[i]<<" "; 
    system("pause");
    return 0;
}