主要解决的问题是:如何从一个无序数组中求出第K大的数
最直接的想法:先排序再输出倒数第K位元素即可。但是时间复杂度为O(nlogn),不如随机选择算法的O(n)。
随机选择算法
核心:一个数组按照从大到小的顺序排序,该数组的第K大元素,一定是在从右往左数的第K位(1开始数)。
算法思路:分治+快排+递归
在区间a【left,right】内,找一个基准元素,让其右边都大于他,让其左边都小于等于他,不妨令该基准元素的下标为:p,他就是从右往左数的第right-p+1,令M = right-p+1,比较M和K。
1.如果相等,那么a【p】就是第K大的元素
2.如果M>k,就说明,右边的元素多了,还要继续往右边区寻找第K大的元素,新区间缩小为【p+1,right】
3.如果M<k,说明右边的元素少了,那么就应该往左边去找K-M大的元素,新的区间缩小为【left,p-1】
举个例子
a [10] = {4,-1,7,9,10,11,58,2,3,6}.
现在求数组中第三大的元素。
1.当某一次划分之后,此时区间为【3,9】,数组为{2,-1,3,6,4,7,58,9,10,11},p为5,M为9-5+1=5,他右边有4个元素,加上自己就是5个元素,大于K,所以因该区现在的右边区间【6,9】找第K大的元素。
代码如下
#include <iostream>
#include <cstdlib>
#include <ctime>
#include <algorithm>
using namespace std;
int randpartition(int *a,int left,int right){
//把区间内所有元素以K为基准分边
int p = (round(1.0 * rand()