这篇文章主要介绍了C++如何实现的O(n)复杂度内查找第K大数,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
具体如下:
题目:是在一组数组(数组元素为整数,可正可负可为0)中查找乘积最大的三个数,最后输出最大乘积。
从题目我们知道只有两种结果存在:
1)三个最大的正整数相乘;
2)一个最大的正整数和两个最小的负数相乘。
所以我们需要找出数组中最大的三个数的乘积m,然后与数组中最小的两个数相乘再与最大的数相乘的结果n,然后比较m,n,选出最大的数即为最终的结果。
参考代码:https://www.jb51.net/article/121189.htm
实现代码:
#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++如何实现的O(n)复杂度内查找第K大数”这篇文章对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,更多相关知识等着你来学习!
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。