leetcode4

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3] nums2 = [2]

则中位数是 2.0 示例 2:

nums1 = [1, 2] nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

package leetcode;

public class leetcode4 {


    static  public double FindMedianSortedArrays(int[] nums1, int[] nums2)
    {
        int length1 = nums1.length;
        int length2 = nums2.length;
        int length = length1 + length2;
        int[] nums = new int[length];


        for (int i = 0 ,i1 = 0 , i2 = 0;i < length; ++i)
        {
            if (i1 == length1)
            {
                nums[i] = nums2[i2++];
            }
            else if (i2 == length2)
            {
                nums[i] = nums1[i1++];
            }
            else
            {
                if (nums1[i1] < nums2[i2])
                {
                    nums[i] = nums1[i1++];
                }
                else
                {
                    nums[i] = nums2[i2++];
                }
            }
        }

        if (length % 2 != 0) {
            return nums[length / 2];
        }
        else
        {
            int l = length / 2;
            return (nums[l - 1] + nums[l]) / 2.0;
        }
    }

    public static void main(String[] args){

        int[] a  = {1,3};
        int[] b = {2};
        System.out.println(FindMedianSortedArrays(a,b));

    }
}