力扣第四题——寻找两个正序数组的中位数

1.问题描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

来源：力扣（LeetCode）链接：https://leetcode.cn/problems/median-of-two-sorted-arrays

2.实例

实例一：

输入：nums1=[1,2],nums2=[3,4]

输出：2.00000

解释：合并数组=[1,2,3,4],中位数（2+3）/ 2=2.5

实例二：

输入：nums1=[1,1],nums2=[1,2]

输出：1.00000

解释：合并数组=[1,1,1,2],中位数（1+1）/ 2=1

3.解法

这道题目其实并不难，给了两个数组，我的思路就是首先将两个数组合并（注意这里一定不要去重，我的例子跟原题的例子不一样，当初没有想到这一点所以一直通过不了），然后排序，再根据元素个数的奇偶判断中位数。

为了方便数组合并和排序我选择用ArrayList来存储两个数组的元素，用了两个for循环将两个数组的元素都添加到list中去，用list.sort方法将数组升序排序。排完序后再根据元素个数判断中位数。

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        ArrayList<Integer> list = new ArrayList<>();
        for (int i:nums1) {
            //将集合nums1中的元素添加到list中
                list.add(i);
        }
        for (int i:nums2) {
            //将集合nums2中的元素添加到list中
                list.add(i);
        }
        list.sort(new Comparator<Integer>(){
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1 - o2;//升序排序
            }
        });
        if (list.size() %2!=0){
            //当元素的个数为奇数时最容易，就是中间那个数。
            return (double) list.get(list.size() /2);
        }
        int size = list.size();
        double num1=list.get(size/2-1);
        double num2 = list.get(size/2);
        return (num1 + num2)/2.0;//这里需要注意list中存储的时int类型，如果不用2.0的话得到的就是一个整数。
        }
}

虽然这题的标签时困难，但是题目本省并不难，困难的是将时间复杂度降低到O(log(m+n))。很明显我的这个方式是错误的，但是懒得费脑子了哈哈。