关注
STL里面的nth_element和partial_sort均可以高效解决此问题,最高效的是nth_element,基本思想是快速排序。具体可以参考本人博客:http://blog.csdn.net/bxw1992/article/details/76695461 void nth_element(vector<int> &vec, int num) 将最小(或最大)的num个数放在数组的开始处。注意num个数是无序的。
主要是利用快速排序的切分操作,源码针对枢轴的选取了优化措施(取待处理区间首、中间、尾3个值中的中间值作为枢轴,防止切分操作退化),这里为了简化,没有对枢轴的选取进行优化。
根据切分函数的返回值,判断是否达到了找出了num个满足要求的数,如果不满足,判断下一处理区间。
#include<iostream>
#include<vector>
using namespace std;
void swap(int &a, int &b)
{
int temp;
temp = a;
a = b;
b = temp;
}
int partition(vector<int> &vec,int low,int hi)
{
int mid = (hi - low) / 2 + low;
int pivot = vec[low];
int i = low + 1;
int j = hi;
while (true)
{
while (i < hi && vec[i] < pivot) i++;
while (j>low && vec[j]>pivot) j--;
if (i >= j) break;
swap(vec[i], vec[j]);
i++;
j--;
}
swap(vec[low],vec[j]);
return j;
}
void nth_element(vector<int> &vec, int num)
{
int len = vec.size();
int low = 0;
int hi = len - 1;
while (low<hi)
{
int j = partition(vec,low,hi);
if (j == num - 1) return;
else if (j < num - 1) low = j+1;
else hi = j - 1;
}
}
void myprint(const vector<int> a)
{
for (int i = 0; i < a.size(); i++)
{
cout << a[i] << " ";
}
cout << endl;
}
int main()
{
vector<int> data = { 11,5, 4, 7, 6, 8, 1,10 };
myprint(data);
nth_element(data,2);
myprint(data);
system("pause");
return 0;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# AI面会问哪些问题? #
23281次浏览 456人参与
# 中国电信笔试 #
30258次浏览 278人参与
# 米连集团26产品管培生项目 #
12826次浏览 284人参与
# 你的实习产出是真实的还是包装的? #
18032次浏览 325人参与
# 从事AI岗需要掌握哪些技术栈? #
7018次浏览 222人参与
# 春招至今,你的战绩如何? #
57290次浏览 518人参与
# 开放七大实习专项,百度暑期实习值得冲吗 #
13569次浏览 202人参与
# 找AI工作可以去哪些公司? #
6973次浏览 166人参与
# 你做过最难的笔试是哪家公司 #
27567次浏览 164人参与
# 哪些公司真双非友好? #
69002次浏览 286人参与
# 阿里笔试 #
174470次浏览 1291人参与
# 面试被问期望薪资时该如何回答 #
382347次浏览 2163人参与
# 晶盛机电求职进展汇总 #
35188次浏览 318人参与
# 一张图晒出你司的标语 #
3637次浏览 67人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
339145次浏览 2158人参与
# 面试尴尬现场 #
220532次浏览 860人参与
# 沪漂/北漂你觉得哪个更苦? #
8728次浏览 183人参与
# ___岗狗都不干,我干! #
77741次浏览 309人参与
# AI时代,哪个岗位还有“活路” #
10293次浏览 317人参与
# 长得好看会提高面试通过率吗? #
21003次浏览 245人参与
# 春招你拿到offer了吗 #
828512次浏览 9975人参与
# HR最不可信的一句话是__ #
5306次浏览 107人参与
