Leetcode NO.4 Median of Two Sorted Arrays

本题题目要求如下:

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)).

题目的要求并不复杂,但是本题的难度还是相当可观的,应该算是hard里面偏简单的题,我第一次虽然有思路,但是确实没有写出来,有以下几点需要注意:  median意义:如果有奇数个数字,则结果应该为最中间的数字,如果有偶数个数字,结果应该是中间两个数的平均数。

然后本题就被简化成寻找第k小的数,算法如下:

1,确保数组a的大小是小于数组b的,如果不符合,则交换顺序

2,比较a第(k/2)小的数,和b第(k/2)小的数字,如果前者小,则可以抛弃所有a的前(k/2)个元素,这些必然比我们寻找的数字小,然后寻找新组成的a,b的第(k/2)小的元素,(k/2)就是新的k

3,如果有特殊情况,比如a的元素书目小于k/2,那直接丢弃a数组(设长度为m),需找b数组的地(k-m)小的元素!

注意:大体算法就是这样,具体实现需要小心,毕竟是一道hard题

代码如下:

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int a[m];
        int b[n];
        copy(nums1.begin(), nums1.end(), a);
        copy(nums2.begin(), nums2.end(), b);
        int total = m + n;
        
        if (total % 2 != 0) {
            return findKthSmallest(a, m, b, n, total / 2 + 1);
        }
        else {
            return findKthSmallest(a, m, b, n, total / 2 + 1) / 2 + findKthSmallest(a, m, b, n, total / 2) / 2;
        }
    }
private:
    double findKthSmallest(int a[], int m, int b[], int n, int k) {
        if (m > n) {
            return findKthSmallest(b, n, a, m, k);
        }
        if (m == 0) {
            return b[k-1];
        }
        if (k == 1) {
            return min(a[0], b[0]);
        }
        
        int pa = min(k / 2, m);
        int pb = k - pa;
        if (a[pa-1] < b[pb-1]) {
            return findKthSmallest(a + pa, m - pa, b, n, k - pa);
        }
        else if (a[pa-1] > b[pb-1]) {
            return findKthSmallest(a, m, b + pb, n - pb, k - pb);
        }
        else {
            return a[pa-1];
        }
    }
};



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值