关注
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;
}
查看原帖
点赞 评论
相关推荐
02-25 16:17
门头沟学院 Java 点赞 评论 收藏
分享
查看21道真题和解析 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# Agent面试会问什么? #
21588次浏览 781人参与
# 哪些公司一直卡在简历筛选 #
99404次浏览 300人参与
# 厦门银行科技岗值不值得投 #
18215次浏览 419人参与
# 给工作过的公司写一条大众点评,你会怎么写? #
7251次浏览 113人参与
# 运营每日一题 #
134297次浏览 928人参与
# 烂工作和没工作哪个更痛苦? #
15372次浏览 246人参与
# 拼多多工作体验 #
54808次浏览 371人参与
# 一人分享一个skill #
5422次浏览 153人参与
# 我们是不是被“优绩主义”绑架了? #
43739次浏览 538人参与
# 春招至今,你收到几个面试了? #
59539次浏览 837人参与
# 央国企投递记录 #
186954次浏览 1710人参与
# 面试体验最好和最差的公司 #
18316次浏览 140人参与
# 机械人面试中的常问题 #
267180次浏览 3692人参与
# 你觉得专业和学校哪个对薪资影响最大 #
103866次浏览 614人参与
# 我是XXX,请攻击我最薄弱的地方 #
71791次浏览 475人参与
# AI替代不了什么? #
10189次浏览 158人参与
# 现在入门AI首先要做什么? #
3548次浏览 92人参与
# 你收到了团子的OC了吗 #
1583055次浏览 11849人参与
# 神州数码信息服务股份有限公司笔试 #
3451次浏览 21人参与
# 你被哪些公司挂了? #
172534次浏览 967人参与
# 面试尴尬现场 #
223998次浏览 868人参与