带权中位数的求解

问题描述

设有n个互不相同的元素x1,x2,…xk，每个元素xi带有一个权值wi，且∑ni=1wi=1。若元素xk满足∑xi<xk wi≤12且∑xi>xk wi≤12，则称元素xk为x1,x2,…,xn的带权中位数。请编写一个算法），能够在最坏情况下用O(n)时间找出n个元素的带权中位数

分析

元素并非事先排序好的，使用类似快速排序的算法，编写函数，选择基准元素，将数组分为三个部分：大于基准元素部分，基准元素，小于基准元素部分。分别计算权重和，在权重和大于0.5的部分继续调用函数。
但时间复杂度仍不满足要求，最坏情况时间复杂度是O(n^2).因此欢迎有更优解法的老哥指正。

代码

#include <iostream>
using namespace std;

void median(double left_sum, double right_sum, int left, int right, double member[], double weight[])
{
    int reference = member[(left+right)/2];
    int i = left, j = right;
    double lsum = 0, rsum = 0;
    while (i < j)
    {
        while ((member[i] < reference)&&(i<j))
        {
            
            lsum += weight[i];
            i++;
        }
        while ((member[j] > reference)&&(i<j))
        {
            
            rsum += weight[j];
            j--;
        }
        if(i<j)
        {
        double temp_member,temp_weight;
        temp_member=member[i];
        temp_weight=weight[i];
        member[i]=member[j];
        weight[i]=weight[j];
        member[j]=temp_member;
        weight[j]=temp_weight;

    }


//用于调试，每调用一次median显示一次其中参数
    // if(i!=j) cout<<"Something wrong"<<endl;
    // cout<<"i="<<i<<"   "<<member[i]<< endl;
    // int z;
    // for(z=left;z<=right;z++)
    // {
    //     cout<<member[z]<<' ';
    // }
    // cout<<endl;
    // for(z=left;z<=right;z++)
    // {
    //     cout<<weight[z]<<' ';
    // }
    // cout<<endl;
    // cout<<"left: "<<lsum<<endl;
    // cout<<"right:"<<rsum<<endl;


    if(((left_sum+lsum)<=0.5)&&((right_sum+rsum)<=0.5))
    {
        cout<<member[i]<<endl;
    }
    else if((left_sum+lsum)>0.5)
    {
        median(left_sum,right_sum+rsum+weight[i],left,i-1,member,weight);
    }
    else 
    {
        median(left_sum+lsum+weight[i],right_sum,i+1,right,member,weight);
    }
}

int main()
{
    int n;
    double *p=nullptr,*q=nullptr;
    cin>>n;
    p=new double[n]();
    if(!p) cout<<"error1"<<endl;
    q=new double[n]();
    if(!q) cout<<"error2"<<endl;
    int i;
    for(i=0;i<n;i++)
    cin>>p[i];
    for(i=0;i<n;i++)
    cin>>q[i];
    median(0,0,0,n-1,p,q);
    return 0;
}

用例

//输入：
10
719 449 446 981 431 993 919 389 549 453
0.01757775 0.02028202 0.16863048 0.07320842 0.16283562 0.16167665 
0.14970060 0.04095036 0.12806645 0.07707166
//输出
549