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

#include 
#include 
//分区
intpartition(std::vector&vec,intstart,intend) {
 intvalue=vec[end];
 inttail=start-1;
 for(inti=start;i&vec,intstart,intend,intk) {
 //快排思想，进行分区，快排复杂度为O(nlgn),但取最值只比较分区的一个区间，所以为O(n)
 intnow = partition(vec,start,end);
 if(k < now)
  returnsolve(vec,start,now-1,k);
 elseif(k > now)
  returnsolve(vec,now+1,end,k);
 else
  returnvec[now];
}
intmain() {
 intn;//要比较的数的个数
 while(std::cin>>n) {
  std::vector vec_i(n,0);//使用vector存储n个数
  for(inti = 0; i < n; ++i) {
   std::cin>>vec_i[i];
  }
  intk;
  //最大的数，index为n-1
  k = n - 1;
  longlongx1 = solve(vec_i,0, n-1,k);
  //次大的数，index为n-2
  k = n - 2;
  longlongx2 = solve(vec_i,0, n-2,k);
  //第三大的数
  k = n - 3;
  longlongx3 = solve(vec_i,0, n-3,k);
  longlongAns = x1 * x2 * x3;//最大的三个数的乘积
  if(n > 3) {
   //最小的数，index为0
   k = 0;
   longlongy1 = solve(vec_i,0, n-1,k);
   //次小的数，index为1
   k = 1;
   longlongy2 = solve(vec_i,0, n-2,k);
   Ans = std::max(Ans, y1*y2*x1);//两者比较取最大
  }
  std::cout<