码迷,mamicode.com
首页 > 编程语言 > 详细

随机选择算法

时间:2020-08-18 13:34:41      阅读:63      评论:0      收藏:0      [点我收藏+]

标签:sele   oid   sum   数组排序   art   选择算法   end   ace   select   

问题:

如何从一个无序的数组中求出第K大的数。

 

思路一:

对数组排序,直接取第K个元素,O(nlogn)

 

思路二:

选择随机选择算法,O(n)

 

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=100010;

int A[maxn],n;//A存放所有整数,n为其个数 
//选取随机主元,对区间[left,right]进行划分
int randPartition(int A[],int left, int right){
    //生成[left,right]内的随机数p
    int p=(round(1.0*rand()/RAND_MAX*(right-left)+left));
    swap(A[p],A[left]);//交换A[p]和A[left]
    //以下为原先Partition函数的划分过程
    int temp=A[left];
    while(left<right){//只要left与right不相遇
        while(left<right&&A[right]>temp){
            right--; 
        }
        A[left]=A[right];//将A[right]挪到A[left]
        while(left<right&&A[left]<=temp){
            left++;
        }    
        A[right]=A[left];
    } 
    A[left]=temp;
    return left;
}

//随机选择算法,从A[left,right]中找到第K大的数,并进行切分
void randSelect(int A[],int left,int right,int K){
    if(left==right){
        return;
    }
    int p=randPartition(A,left,right);    
    int M=p-left+1;
    if(K==M){
        return ; 
    }
    if(K<M){
        randSelect(A,left,p-1,K);
    }
    else{
        randSelect(A,p+1,right,K-M);
    }
} 

int main(){
    srand((unsigned)time(NULL));
    int sum=0,sum1=0;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>A[i];
        sum+=A[i];
    }
    randSelect(A,0,n-1,n/2);
    for(int i=0;i<n/2;i++){
        sum1+=A[i];
    }
    cout<<(sum-sum1-sum1)<<endl;
    return 0;
}

 

随机选择算法

标签:sele   oid   sum   数组排序   art   选择算法   end   ace   select   

原文地址:https://www.cnblogs.com/ak918xp/p/13511486.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!