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

算法思想

给出两个有序的数组，要求其中的中位数，显然，当把两个数组合并成一个有序的数组之后，再求中位数就变得非常简单了，因此，我们需要将这两个有序的数组归并成一个有序的数组，而归并算法的时间复杂度恰好为O(n)
归并算法描述：需要创建一个新的数组，用来保存归并后的有序数组，我们需要使用两个指针变量i,j,分别指向两个有序数组的头部，然后让i,j所指向的元素进行比较，将小的元素放到我们的新数组中，依次这样做，直到所有的元素都放入到新数组中。

代码描述

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m=nums1.length;		//获取数组1的长度
		int n=nums2.length;
		int sum=m+n;
		int nums[]=new int[sum+10];	//创建一个新的数组
		//将两个数组归并起来成为一个更长的数组
		int i=0,j=0,k=0;		//i,j,k指针分别指向三个数组的首端
		while((i<m)&&(j<n)){
			//当数组nums1中的元素更小，就将其放入到新数组中，指针后移一位
			while((i<m&&j<n)&&(nums1[i]<nums2[j])){
				nums[k++]=nums1[i++];
				
			}
			//当数组nums2中的元素更小，就将其放入到新数组中，指针后移一位
			while((i<m&&j<n)&&(nums1[i]>=nums2[j])){
				nums[k++]=nums2[j++];
			}
		}
		//经过上一轮循环中，可能其中有一个数组中的元素还未全部放入到新数组中
		while(i<m){
			nums[k++]=nums1[i++];
		}
		while(j<n){
			nums[k++]=nums2[j++];
		}
		float result;
		if(sum%2==0){
		//注意，要想得到浮点数的结果，除数就必须为小数
			result=(float) ((nums[sum/2-1]+nums[sum/2])/(2.0));
		}else{
			result=nums[sum/2];
		}
		return result;
    }    
}