分治与递归——带权中位数

问题描述

设有n个互不相同的元素x1,x2,…, xⁿ，每个元素x_i带有一个权值w_i，且${\sum }_{i=1}^n{w}_i=1$, 若元素x_k满足${\sum }_{x_i<x_k}{W}_i\le 0.5$，且${\sum }_{x_i>x_k}{W}_i\le 0.5$，则称元素x_k为$x_1,x_2\cdots x_n$的带权中位数。请编写一个算法，能够在最坏情况下用O(n)时间找出n个元素的带权中位数。

问题分析

该问题可以转化为规模较小，问题相同的子问题，所以我们采用分支和递归的思想。

算法设计

本题使用快速排序即双指针的思想。

以分界值为标准，将数组划分为前后两部分，并分别统计前后两部分的权重和，

1.若前一半权重和小于0.5且后一半权重和小于0.5，则该分界值为所求。

2.若前一半权重和大于0.5且后一半权重和小于0.5，则递归运行数组的前一部分，直到出现1的情况。

3.若前一半权重和小于0.5且后一半权重和大于0.5，则递归运行数组的后一部分，直到出现1的情况。

算法实现

#include<iostream>
#include<vector>
#include<stdlib.h>
using namespace std;

struct ele
{
    int x;
    double w;
};
int n, ans = -1;

void binarySearch(vector<ele>& a, int left, int right, double leftsum, double rightsum)
{
    if(right<left) return;
    if(right==left) {ans = a[left].x; return;}
    int l = left+1, r = right, gap = right-left;
    double rsum = rightsum, lsum = leftsum;
    int pivot = left; 
    while(l<=r)
    {
        while(a[pivot].x < a[r].x) { rsum += a[r].w ;r--; } 
        while(a[pivot].x > a[l].x) { lsum += a[l].w; l++; }
        if(l<r) swap(a[l], a[r]);
    }
    swap(a[pivot], a[r]);
    if(rsum<0.5 && lsum<0.5)
    {
        ans = a[r].x;
        return;
    }
    else if(rsum <0.5 && lsum>0.5)
    {
        binarySearch(a, left, r-1, leftsum, rsum);
    }
    else if(rsum > 0.5 && lsum < 0.5)
    {
        binarySearch(a, r+1, right, lsum, rightsum);
    }
}

int main()
{
    cin>>n;
    ele m; m.w = 0; m.x = 0;
    vector<ele> a(n, m);
    for(int i = 0; i < n; i++) cin>>a[i].x;
    for(int i = 0; i < n; i++) cin>>a[i].w;
    binarySearch(a, 0, n-1, 0, 0);
    cout<<endl<<ans<<endl;
    return 0;
}