[leetcode] 4. 寻找两个有序数组的中位数

给定两个大小为 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

 

要求时间复杂度为log(m+n),意思就是只能单层循环一次两个数组，我的思路是从两个数组中比较大小取数放入一个list中，生成一个从小到大排序的list，然后直接按情况取中位数，如果list里有偶数个数则返回{（个数/2）+[(个数/2)-1]}/2，如果list的个数是奇数则直接返回中位数。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        List<Double> l = new LinkedList<>();
        int nums1_length = nums1.length;
        int nums2_length = nums2.length;
        boolean flag = true;
        int m = (nums1_length+nums2_length)/2;
        if((nums1_length + nums2_length)%2!=0){
            flag = false;
        }
        int i=0,j=0;
        while(i<nums1_length&&j<nums2_length){
            if(nums1[i]<nums2[j]){
                l.add((double)nums1[i]);
                i++;
            }else if(nums1[i] == nums2[j]){
                l.add((double)nums1[i]);
                l.add((double)nums2[j]);
                i++;
                j++;
            }else{
                l.add((double)nums2[j]);
                j++;
            }
        }
        
        if(i<nums1_length){
            while(i<nums1_length){
                l.add((double)nums1[i]);
                i++;
            }
        }
        
        if(j<nums2_length){
            while(j<nums2_length){
                l.add((double)nums2[j]);
                j++;
            }
        }
        double middle;
        if(flag){
            middle = (l.get(m)+l.get(m-1))/2;
        }else{
            middle = l.get(m);
        }
        
        return middle;
    }
    
}