4. Median of Two Sorted Arrays

在寻找两个已经排好序的数组的第k小的元素的时候参考了下面一篇博客
http://blog.csdn.net/beiyeqingteng/article/details/7533304/

class Solution {
private:
    int findK(vector<int>& nums1,int pa,vector<int>& nums2,int setpa,int k)
    {
        int pb=k-1-pa;
        int PA_1=(pa==0)?INT_MIN:nums1[pa-1];
        int PB_1=(pb==0)?INT_MIN:nums2[pb-1];
        int PA=(pa==nums1.size())?INT_MAX:nums1[pa];
        int PB=(pb==nums2.size())?INT_MAX:nums2[pb];
        if(PA>=PB_1&&PB>=PA) return PA;
        if(PB>=PA_1&&PA>=PB) return PB;
        if(PA>PB)
        {
            pa=((k-1-(pa-setpa))>nums2.size())?(k-1-nums2.size()):(pa-setpa);
            return findK(nums1,pa,nums2,(setpa+1)/2,k);
        }
        else 
        {
            pa=((pa+setpa)>nums1.size())?nums1.size():pa+setpa;
            return findK(nums1,pa,nums2,(setpa+1)/2,k);
        }
    }
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
    {
        int l1=nums1.size();
        int l2=nums2.size();
        int l=l1+l2;
        if(l%2==0)
        {
            int k=l/2;
            int pa=min((int)nums1.size(),k-1);
            int target1=findK(nums1,pa,nums2,(pa+1)/2,k);
            k=l/2+1;
            pa=min((int)nums1.size(),k-1);
            int target2=findK(nums1,pa,nums2,(pa+1)/2,k);
            return (target1+target2)/2.0;
        }
        else
        {
            int k=(l+1)/2;
            int pa=min((int)nums1.size(),k-1);
            return findK(nums1,pa,nums2,(pa+1)/2,k)*1.0;
        }
    }

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

hebastast

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值