C++实现的O(n)复杂度内查找第K大数算法示例

时间:2021-05-20

本文实例讲述了C++实现的O(n)复杂度内查找第K大数算法。分享给大家供大家参考,具体如下:

题目:是在一组数组(数组元素为整数,可正可负可为0)中查找乘积最大的三个数,最后输出最大乘积。

从题目我们知道只有两种结果存在:
1)三个最大的正整数相乘;
2)一个最大的正整数和两个最小的负数相乘。

所以我们需要找出数组中最大的三个数的乘积m,然后与数组中最小的两个数相乘再与最大的数相乘的结果n,然后比较m,n,选出最大的数即为最终的结果。

参考代码:

实现代码:

#include <iostream>#include <algorithm>//分区int partition(std::vector<int>&vec,int start,int end) { int value=vec[end]; int tail=start-1; for(int i=start;i<end;++i){ if(vec[i]<value){ tail++; std::swap(vec[i],vec[tail]); } } tail++; std::swap(vec[tail],vec[end]); return tail;}long long solve(std::vector<int>&vec,int start,int end,int k) { //快排思想,进行分区,快排复杂度为O(nlgn),但取最值只比较分区的一个区间,所以为O(n) int now = partition(vec,start,end); if(k < now) return solve(vec,start,now-1,k); else if(k > now) return solve(vec,now+1,end,k); else return vec[now];}int main() { int n;//要比较的数的个数 while(std::cin>>n) { std::vector<int> vec_i(n,0);//使用vector存储n个数 for(int i = 0; i < n; ++i) { std::cin>>vec_i[i]; } int k; //最大的数,index为n-1 k = n - 1; long long x1 = solve(vec_i,0, n-1,k); //次大的数,index为n-2 k = n - 2; long long x2 = solve(vec_i,0, n-2,k); //第三大的数 k = n - 3; long long x3 = solve(vec_i,0, n-3,k); long long Ans = x1 * x2 * x3;//最大的三个数的乘积 if(n > 3) { //最小的数,index为0 k = 0; long long y1 = solve(vec_i,0, n-1,k); //次小的数,index为1 k = 1; long long y2 = solve(vec_i,0, n-2,k); Ans = std::max(Ans, y1*y2*x1);//两者比较取最大 } std::cout<<Ans; } return 0;}

希望本文所述对大家C++程序设计有所帮助。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章