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

来源:互联网 发布:女生一身淘宝爆款low 编辑:程序博客网 时间:2024/05/20 06:38

问题难度: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;        }    }};



原创粉丝点击