Leetcode问题解答:4. Median of Two Sorted Arrays

问题难度:Hard

源链接:https://leetcode.com/problems/median-of-two-sorted-arrays/description/

描述:

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).


思考与解读:



该问题要求找出给定的两个有序数组中的中位数,并且有一定的时间复杂度限制,不能把数组合并再排序这样来获得答案。
观察发现对于两个有序数组的情况和归并排序中merge的过程有一定相似性,可以通过merge的思路来处理这两个数组,
使用两个标记和一个计数器,两个标记在两个数组上递增,且只选择对应值相比较小的标记递增,计数器显示到总大小的二分之一停止,这样便获得了中位数且
对于数组中的每个数至多访问一次。

代码如下:

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int* result = new int[nums1.size()+nums2.size()+1];
        int total = 1;
        int p1 = 0;
        int p2 = 0;
        while((nums1.size()+nums2.size())/2+1 >= total){
            if (p1 >= nums1.size() &&  p2 < nums2.size()) {
                result[total] = nums2[p2];
                p2++;
            } else if (p1 < nums1.size() &&  p2 >= nums2.size()) {
                result[total] = nums1[p1];
                p1++;
            } else {
                if (nums1[p1] < nums2[p2]) {
                    result[total] = nums1[p1];
                    p1++;
                } else {
                    result[total] = nums2[p2];
                    p2++;
                }
            }
            total++;
        }
        double a = result[(nums1.size()+nums2.size())/2];
        double b = result[(nums1.size()+nums2.size())/2+1];
        if ((nums1.size()+nums2.size())%2 == 0) {
            return (a+b)/2;
        } else {
            return b;
        }
    }
};



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值