java解法——寻找两个有序数组的中位数

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

思路：求中位数，所以只需将两个数组的一半之数排序即可。因为是两个已经排好序的数组，所以只需依次从头比较，较小的放入新建的数组之中即可。

 

import java.util.*;


public class Solution {
    /**
     * 
     * @param A int整型一维数组 
     * @param B int整型一维数组 
     * @return double浮点型
     */
    public double findMedianSortedArrays (int[] A, int[] B) {
       int sum = A.length + B.length;
       int temp = 0;
       int[] arr = new int[sum];
       int i=0,j=0;
       double result;
        
       //思路：求中位数，所以只需将两个数组的一半之数排序即可。因为是两个已经排好序的数组，所以只需依次从头比较，较小的放入新建的数组之中即可。
       while(temp <= sum/2){
           
           //防止数组溢出
           if(i<A.length && j<B.length){
               
               if(A[i] < B[j]){
                   arr[temp++] = A[i++];
               }else{
                   arr[temp++] = B[j++];
               }
           }else if(i >= A.length){//A数组已经到达末尾了 A.length-1的时候 A就已经遍历完成了
               arr[temp++] = B[j++];
           }else{//数组B 遍历完成 到达末尾了 相当于j>=B.length
               arr[temp++] = A[i++];
           }
       }
        
       if(sum%2 == 0){//2个数组合并后 偶数位个长度
           result = ((double)(arr[sum/2] + arr[sum/2 - 1])) / 2;
       }else{//2个数组合并后 奇数位个长度
           result = (double)arr[sum/2];
       }
        
       return result;
    }
}