LeetCode04 寻找两个有序数组的中位数 Java版

1. 题目

给定两个大小为m和n的有序数组nums1 和 nums2
请你找出这两个有序数组的中位数, 并且要求算法的时间复杂度为log(m+n)
假设: nums1 和nums2不会同时为空

2. 初步实现

import org.apache.commons.lang.ArrayUtils;

/**
 * 给定两个大小为m和n的有序数组nums1 和 nums2
 * 请你找出这两个有序数组的中位数, 并且要求算法的时间复杂度为log(m+n)
 * 假设: nums1 和nums2不会同时为空
 */
public class Solution {
    public static void main(String[] args) {
        long start = System.currentTimeMillis();
        test();
        long end = System.currentTimeMillis();
        System.out.println("运行耗时" + (end - start) + "ms");
    }

    public static void test() {
        int nums1[] = {1, 2};
        int nums2[] = {3, 4};

        int[] nums3 = ArrayUtils.addAll(nums1, nums2);
        int len = nums3.length;
        int x = len/2;
        if (len % 2 == 0) {
            System.out.println((nums3[x] + nums3[x - 1]) / 2.0);

        } else {
            System.out.println(nums3[x]);
        }
    }
}

发现要运行好久啊~

3. 初步优化

根据网上大神的想法,进行初步优化~

/**
 * 给定两个大小为m和n的有序数组nums1 和 nums2
 * 请你找出这两个有序数组的中位数, 并且要求算法的时间复杂度为log(m+n)
 * 假设: nums1 和nums2不会同时为空
 */
public class Solution {
    public static void main(String[] args) {
        long start = System.currentTimeMillis();
        test();
        long end = System.currentTimeMillis();
        System.out.println("运行耗时" + (end - start) + "ms");
    }

    public static void test() {
        // 两个数组
        int nums1[] = {1, 2};
        int nums2[] = {3, 4};
        // 两个数组分别的下标
        int left = 0;
        int right = 0;
        // 两个数组的元素之和
        int len = nums1.length + nums2.length;
        // 记录循环到的上一个值
        int res_fon = 0;
        // 记录当前循环到的值
        int res = 0;
        int x = len / 2;
        // 判断是奇数还是偶数
        if ((nums1.length + nums2.length) % 2 == 0) {
            // 奇数的话去当前循环的值 + 上一个循环的值, 然后除以2.0
            for (int i = 0; i <= x; i++) {
                // 记录一下上一个值
                res_fon = res;
                // 当 下标<数组的长度 && 当前数组的值<其他数组的值
                if (left < nums1.length && nums1[left] < nums2[right]) {
                    res = nums1[left++];
                } else {
                    res = nums2[right++];
                }
            }
            System.out.println((res + res_fon) / 2.0);
        } else {
            // 偶数的话当前循环到的值就是中位数
            for (int i = 0; i <= x; i++) {
                if (left < nums1.length && nums1[left] < nums2[right]) {
                    res = nums1[left++];
                } else {
                    res = nums2[right++];
                }
            }
            System.out.println(res);
        }
    }
}

4. 再次优化

/**
 * 给定两个大小为m和n的有序数组nums1 和 nums2
 * 请你找出这两个有序数组的中位数, 并且要求算法的时间复杂度为log(m+n)
 * 假设: nums1 和nums2不会同时为空
 */
public class Solution {
    public static void main(String[] args) {
        long start = System.currentTimeMillis();
        int[] nums1 = {1};
        int[] nums2 = {2, 3, 4, 5};
        int len = nums1.length + nums2.length;
        int k = len / 2;
        int p1 = k / 2 - 1;
        int p2 = k / 2 - 1;
        while (k != 0 && p1 < nums1.length - 1 && p2 < nums2.length - 1) {
            if (nums1[p1] > nums2[p2]) {
                k = k / 2;
                p2 = p2 + k;
                p1 = k - 1;
            } else {
                k = k / 2;
                p1 = p1 + k;
                p2 = k - 1;
            }
        }
        if (p1 >= nums1.length - 1) {
            if (len % 2 == 0) {
                System.out.println((nums2[p2 + k] + nums2[p2 + k - 1]) / 2.0);
            } else {
                System.out.println(nums2[p2 + k - 1]);
            }
        } else if (p2 >= nums2.length - 1) {
            if (len % 2 == 0) {
                System.out.println((nums1[p1 + k] + nums1[p1 + k - 1]) / 2.0);
            } else {
                System.out.println(nums1[p1 + k - 1]);
            }
        } else {
            if (len % 2 == 0) {
                System.out.println((nums1[p1] + nums2[p2]) / 2.0);
            } else {
                System.out.println(nums1[p1] < nums2[p2] ? nums2[p2] : nums1[p1]);
            }
        }
        long end = System.currentTimeMillis();
        System.out.println("运行耗时" + (end - start) + "ms");
    }


}