首页 > 其他 > 详细

线性时间选择——第k个数

时间:2021-05-01 12:24:07      阅读:22      评论:0      收藏:0      [点我收藏+]

线性时间选择——选择第k小的数

??采用的算法思想和快速排序十分相似,选取基准进行一次快排,得到一个基准和左侧小于基准右侧大于基准的数组。然后判断基准的位置,如果第k个数在左边便接着向左递归,如果在右侧便接着向右侧递归。

程序实现

#include<iostream>
#include<set>
#include<cmath>
#include<vector>
#include<map>
#include<string>
#include<sstream>

using namespace std;


//进行分割排序,并且返回排序后基准的位置
template <typename  Type>
int Partition(Type a[],int p,int r){
    int i=p,j=r+1;  //这里的j=r+1,之所以奖赏一是因为后面的--j,要先做减法,为了防止最后一个元素称为漏网之鱼
    Type x=a[p];    //这里将数组的起始点下标p指向的元素作为基准
    //将小于x的元素交换到左边区域,将大于x的元素交换到右边区域
    while(true){
        while(a[++i]<x&&i<r);   //先扫描左边的,直到找到比x大的元素停下来
        while(a[--j]>x);        //找到右边比x小的元素停下来
        if(i>=j)break;          //结束条件
        Swap(a[i],a[j])         //将他俩进行交换
    }
    //这里j指向的要小于等于x
    a[p]=a[j];
    a[j]=x;
    return j;
}

//进行随机分割,避免数组整体倒序造成时间复杂度升高
template <typename  Type>
int RandomizedPartition(Type a[],int p,int r){
    int i=random(p, r);
    swap(a[i],a[p]);
    return Partition(a,p,r);
}

/*
    线性时间选择第k个数:
        a[] :数组
        p   :数组起点
        r   :数组终点
        k   :第k个数
*/
template <typename  Type>
Type RandomizedSelect(Type a[],int p,int r,int k){
    //如果找到k,即分割到只有一个数的时候
    if(p==r)    return a[p];
    int i=RandomizedPartition(a,p,r);
    int j=i-p+1;    //用来判断第k个数是在基准的左边还是右边
    if(k<=j)return RandomizedSelect(a,p,i,k);   //如果在左边
    else return RandomizedSelect(a,i+1,r,k-j);  //如果在右边
}

线性时间选择——第k个数

原文:https://www.cnblogs.com/3236676588buladuo/p/14723910.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!