寻找两个正序数组的中位数

1,532

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

前言

  • LeetCode第四题:寻找两个正序数组的中位数,难道:困难
  • 题目描述:给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请找出并返回这两个正序数组的中位数;
  • 这道题给我的感觉其实还没有上次做的-无重复最长字符串-难,基本上提交一次就成功了【狗头/滑稽】;

方式一:并归数组

  • 解题重点两个正序数组,所以提供的数组已经是排好序的,只需要在合并到一起凑成一个大数组即可
public double findMedianSortedArrays (int[] nums1, int[] nums2) {
    int all = nums1.length + nums2.length;
    int[] arr = new int[all];
    int i = 0, i1 = 0, i2 = 0;
    while (i1 < nums1.length && i2 < nums2.length) {
        if (nums1[i1] <= nums2[i2]) {
            arr[i++] = nums1[i1++];
        } else {
            arr[i++] = nums2[i2++];
        }
    }
    while (i1 < nums1.length) {
        arr[i++] = nums1[i1++];
    }
    while (i2 < nums2.length) {
        arr[i++] = nums2[i2++];
    }
    int a = all >> 1;
    if (all % 2 == 0) {
        return (arr[a - 1] + arr[a]) / 2d;
    }
    return arr[a];
}
  • 这种方式是时间复杂度是O(m+n),空间复杂度也是O(m+n);

方式二:查找中位数

  • 感觉比上述的并归方式稍微难写(并归方式一下搞定通过,这种方式思考了很久才做出来的 O.O)
  • 这种方式通过看官方解题介绍里有描述,看了一些其他的解题思路,还是摸索出来了。
public double findMedianSortedArrays (int[] nums1, int[] nums2) {
    int len = nums1.length + nums2.length;
    int i = 0, i1 = 0, i2 = 0, s = len >> 1, value1, value2 = 0;
    do {
        value1 = value2;
        if (i1 < nums1.length && (i2 >= nums2.length || nums1[i1] <= nums2[i2])) {
            value2 = nums1[i1++];
        } else {
            value2 = nums2[i2++];
        }
    } while (s >= ++i);
    return len % 2 == 0 ? (value1 + value2) / 2d : value2;
}
  • 这种方式是时间复杂度可以当做是O(m+n),空间复杂度是O(1);

最后

  • 当然这题还有其他的解法,在其官方介绍里面使用二分查找可以使时间复杂度降低至O(log(m+n));
  • 虚心学习,共同进步 -_-