一个长度为A的无序数组,找到中位数

16 篇文章 0 订阅

如果数组长度是奇数,中位数是排序后的第(n+1)/2个元素;若是偶数,中位数是排序后第n/2个元素。

思路1:

1.1 将前(n+1)/2个元素调整为一个最小堆;
1.2 对后续每一个元素和堆顶比较,如果小于等于堆顶,丢弃之,去下一个元素。如果大于堆顶,用该元素取代堆顶,调整堆,去下一个元素重复1.2步
1.3 当遍历完所有元素之后,堆顶为中位数

思路2:

可以扩展为从无序数组中查找第k大的元素。

利用快速排序的partition函数,任意挑一个元素,以该元素key,划分数组为两部分,key左边元素小于等于key,右边元素大于等于key。在第一次partition后,如果左侧元素个数<k - 1,则在右侧子序列中递归查找;如果左侧元素个数=k-1,则第k大元素即在分点处;如果左侧元素个数>k - 1,则递归地在左侧序列中继续查找。代码如下:
#include <iostream>
#include <vector>

using namespace std;

void swap(vector<int> &arr, int i, int j){
    int tmp = arr[i];
    arr[i] = arr[j];
    arr[j] = tmp;
}

int partition(vector<int> &arr, int low, int high){
    int pivot = arr[low];
    int i = low, j = high;
    while (i <= j) {
        while (i <= j && arr[i] <= pivot) i++;
        while (i <= j && arr[j] >= pivot) j--;
        if (i >= j) break;
        swap(arr, i, j);
    }
    swap(arr, low, j);
    return j;
}

int selectK(vector<int> &arr, int k, int low, int high){
    if (k > high - low + 1) return -1;
    int p = partition(arr, low, high);
    if (p - low < k - 1){
        return selectK(arr, k - p - 1, p + 1 , high);
    }
    else if (p - low == k - 1){
        return arr[p];
    }
    else{
        return selectK(arr, k, low, p - 1);
    }
}

int main(int argc, const char * argv[])
{
	vector<int> tmp;
    tmp.push_back(2);
    tmp.push_back(2);
    tmp.push_back(12);
    tmp.push_back(3);
    
    if (tmp.size() & 1){
        cout << selectK(tmp, (tmp.size() + 1) / 2, 0, tmp.size() - 1);
    }
    else{
        cout << selectK(tmp, tmp.size() / 2, 0, tmp.size() - 1);
    }
    
	system("pause");
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值