寻找两个正序数组的中位数

//给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。 
//
// 进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗? 
//
// 
//
// 示例 1: 
//
// 输入:nums1 = [1,3], nums2 = [2]
//输出:2.00000
//解释:合并数组 = [1,2,3] ,中位数 2
// 
//
// 示例 2: 
//
// 输入:nums1 = [1,2], nums2 = [3,4]
//输出:2.50000
//解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
// 
//
// 示例 3: 
//
// 输入:nums1 = [0,0], nums2 = [0,0]
//输出:0.00000
// 
//
// 示例 4: 
//
// 输入:nums1 = [], nums2 = [1]
//输出:1.00000
// 
//
// 示例 5: 
//
// 输入:nums1 = [2], nums2 = []
//输出:2.00000
// 
//
// 
//
// 提示: 
//
// 
// nums1.length == m 
// nums2.length == n 
// 0 <= m <= 1000 
// 0 <= n <= 1000 
// 1 <= m + n <= 2000 
// -106 <= nums1[i], nums2[i] <= 106 
// 
// Related Topics 数组 二分查找 分治算法 
// 👍 3735 👎 0


//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    /**
     * 使用归并排序,将两个数组合并.时间复杂度O(nlogn)
     * @param nums1
     * @param nums2
     * @return
     */
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int[] temp = new int[nums1.length+nums2.length];
        int i,j,k;
        i = j =k =0;
        while(i<nums1.length && j < nums2.length) {
            if(nums1[i] < nums2[j]) {
                temp[k++]=nums1[i++];
            }else {
                temp[k++]=nums2[j++];
            }
        }
        while(i<nums1.length){
            temp[k++] = nums1[i++];
        }

        while(j<nums2.length){
            temp[k++] = nums2[j++];
        }

        if(temp.length%2==1) {
            return temp[temp.length/2]/1d;
        }else {
            return (temp[temp.length/2-1]+temp[temp.length/2])/2d;
        }
    }

}
//leetcode submit region end(Prohibit modification and deletion)

posted @ 2021-03-01 23:42  澎拜编程  阅读(102)  评论(0)    收藏  举报