Leetcode - Median of Two Sorted Arrays

**
Question:

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

**

My code:

public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null)
            return 0;
        if (nums1.length == 0 && nums2.length == 0)
            return 0;
        int m = nums1.length;
        int n = nums2.length;
        if (m == 0)
            if (n % 2 == 1)
                return nums2[n / 2];
            else
                return ((double)nums2[n / 2] + (double)nums2[n / 2 - 1]) / 2.0;
        else if (n == 0)
            if (m % 2 == 1)
                return nums1[m / 2];
            else
                return ((double)nums1[m / 2] + (double)nums1[m / 2 - 1]) / 2.0;
        
        
        int indexOfNums1 = 0;
        int indexOfNums2 = 0;
        boolean isOutOfRange = false;
        int[] mergeArray = new int[m + n];
        for (int i = 0; i < m + n; i++) {
            
            if (indexOfNums1 < m && (nums1[indexOfNums1] < nums2[indexOfNums2] || isOutOfRange)) {
                mergeArray[i] = nums1[indexOfNums1];
                indexOfNums1++;
            }
            else {
                mergeArray[i] = nums2[indexOfNums2];
                if (indexOfNums2 < n - 1)
                    indexOfNums2++;
                else
                    isOutOfRange = true;
            }
        }
        
        if ((m + n) % 2 == 1)
            return mergeArray[(m + n) / 2];
        else
            return ((double)mergeArray[(m + n) / 2] + (double)mergeArray[(m + n) / 2 - 1]) / 2.0;
        
    }
    
    public static void main(String[] args) {
        Solution test = new Solution();
        int[] a = {100001};
        int[] b = {100000};
        System.out.println(test.findMedianSortedArrays(a, b));
    }
}

My test result:

Leetcode - Median of Two Sorted Arrays_第1张图片
Paste_Image.png

这次的难度是hard,但是我一看到就有了思路,然后花了十分钟写了出来。
这不就是归并排序(Merge Sort)的简单版本么。。。
给两个已排序好的数列,然后将它们合并,获得一个新的数列
来,复习下归并排序。
我自己创造的一个图书。
. . . . . . . . 一个array,用归并排序来处理
. . . . | . . . . 分成两个array,分别进行归并排序
. . | . . | . . | . . 对于每个子array,再次分成两个array,分别进行归并排序。
此时每个部分只有两个数字了, lo指向第一个,头。hi指向最后一个,也是第二个,尾。
于是进行简单的排序。完成了。退回到上步。
. . | . . | . . | . . 此时的子array都是已经排序好了的。于是将 1,2子array归并,3,4子array归并。
. . . . | . . . . 此时的两个子array也是已经排序好了的。于是将它们归并,得到最后的排序结果。
. . . . . . . .

然后归并排序的核心就是这次作业写的。
两个array,然后我先创建一个array长度是他们的长度和。
然后通过移动指针,将两个数组对应位置较小的拷贝进入mergeArray,然后指针再往前一格。这样,扫描(m + n)次就能完成了。
然后处理一些数组越界问题。就差不多。复杂度是 N * Log(N)

**
总结:
归并排序。
数组越界。
**

最近两次作业都是十几分钟写出来了。。。是我太强了吗?
我不觉得。我觉得是评测系统有问题。这两道题目正常人应该都半小时内就能做出来吧。
最近听了荔枝FM的一个广播。额,首先声明,我绝对不是做广告的人。
很感动。然后那么一句话。
**
其实我也是一个普通人啊。
**
打到我内心了。
小说世界里, 泰广,米可,最后能破镜重圆。
现实世界里,会有这般失而复得的事吗?
所以,确定你想要的,然后坚持下去。
读者们肯定看不懂我说的这些话,但你呢,应该看得懂吧。
我愿意为你改变很多,放弃很多。
你也愿意为我忍耐很多,担心很多。
异地异国实在是太难熬,但熬出头了呢。
如果因为那么一件件的事,咱们最终放弃。不知道你会怎么想。
我的话,一定会有新的女朋友,家庭。
但这份对你的遗憾,会陪伴我,直到进入棺材。

Anyway, Good luck, Richardo!

My code:

public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null) {
            return -1;
        }
        else if (nums2.length < nums1.length) {
            return findMedianSortedArrays(nums2, nums1);
        }
        else if (nums2.length == 0) {
            return -1;
        }
        
        
        int m = nums1.length;
        int n = nums2.length;
        
        int half = (m + n + 1) / 2;
        int imin = 0;
        int imax = m;
        while (imin <= imax) {
            int i = (imax + imin) / 2;
            int j = half - i;
            if (i - 1 >= 0 && j < n && nums1[i - 1] > nums2[j]) {
                imax = i - 1;
            }
            else if (j - 1 >= 0 && i < m && nums2[j - 1] > nums1[i]) {
                imin = i + 1;
            }
            else {
                int left = 0;
                if (i == 0) {
                    left = nums2[j - 1];
                }
                else if (j == 0) {
                    left = nums1[i - 1];
                }
                else {
                    left = Math.max(nums1[i - 1], nums2[j - 1]);
                }
                
                if ((m + n) % 2 == 1) {
                    return left;
                }
                
                int right = 0;
                if (i == m) {
                    right = nums2[j];
                }
                else if (j == n) {
                    right = nums1[i];
                }
                else {
                    right = Math.min(nums1[i], nums2[j]);
                }
                
                return (left + right) / 2.0;
            }
        }
        return -1;
    }
    
    public static void main(String[] args) {
        Solution test = new Solution();
        int[] nums1 = new int[]{1,2};
        int[] nums2 = new int[]{3,4,5,6};
        double ret = test.findMedianSortedArrays(nums1, nums2);
        System.out.println(ret);
    }
}

reference:
https://discuss.leetcode.com/topic/4996/share-my-o-log-min-m-n-solution-with-explanation/2

这道题目实在是太恶心了。弄了一个多小时。
corner case 太多了。。。

看下讲解,思路清晰了很多。但是还是调了很久很久。
最大的问题是,得把短的作为切割的第一数组,即, i

因为如果切长的,作为i
那么 j很有可能变成负数,会出现错误。
如果切短的,就不会用这个问题。

还有个注意的地方就是:
half = (m + n + 1) / 2;
这样保证,如果 m + n 是偶数,
那么左右部分个数相等。
如果是奇数,那么左边会多一个。

而且imax = m 而不是 m - 1,真的求中点。
如果m是奇数,就是正中点。如果m是偶数,是中间两个数右边那个。
这些正好都是我们需要的特性。

实在是太恶心了。

加油!

Anyway, Good luck, Richardo! -- 09/01/2016

你可能感兴趣的:(Leetcode - Median of Two Sorted Arrays)