LintCode 81:Data Stream Median

维护了一个排好序的vector,每次用二分法插入数,插入的平均时间复杂度为O(logN)。


class Solution {
    void InsertAndSort(vector<int>& sortVector,int num){
        int low=0;
        int high=sortVector.size()-1;
        while(low<high){
            if(num>=sortVector[low]&&num<=sortVector[low+1]){
                sortVector.insert(sortVector.begin()+low+1,num);
                return;
            }
            if(num>sortVector[(low+high)/2]){
                if(low==(low+high)/2)
                    low=(low+high)/2+1;
                else
                    low=(low+high)/2;
            }

            else
                high=(low+high)/2;
        }

        //when sortVector is empty.
        if(high==-1)
            sortVector.push_back(num);
        //when num should be in the front or at the back.
        else{
            if(num<=sortVector[0])
                sortVector.insert(sortVector.begin(),num);
            else if(num>=sortVector.back())
                sortVector.insert(sortVector.end(),num);
        }

        return;
    }
public:
    /**
     * @param nums: A list of integers.
     * @return: The median of numbers
     */
    vector<int> medianII(vector<int> &nums) {
        // write your code here
        vector<int> sortVector;
        vector<int> result;
        for(int i=0;i<nums.size();i++){
            InsertAndSort(sortVector,nums[i]);
            result.push_back(sortVector[i/2]);
        }
        return result;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值