Sandwichsauce的博客

野生母程序猿一只

LeetCode|4.两个排序数组的中位数(Java)

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 

请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

示例 1:

nums1 = [1, 3]
nums2 = [2]

中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

中位数是 (2 + 3)/2 = 2.5
正确代码:A4两个排序数组的中位数.java
package LeetCode;

import java.util.ArrayList;

public class A4两个排序数组的中位数 {
	public double findMedianSortedArrays(int[] nums1, int[] nums2) {
		int n1 = nums1.length;
		int n2 = nums2.length;
		int i = 0;
		int j = 0;
		int k = 0;
		int[] nums = new int[n1 + n2];
		while (n1 != 0 && n2 != 0) { // 公共长度先对比,小的放在新的数组里
			nums[k] = (nums1[i] < nums2[j]) ? nums1[i] : nums2[j];
			if (nums1[i] < nums2[j]) {
				i++;
				k++;
				n1--;
			} else {
				j++;
				k++;
				n2--;
			}
		}
		if (n1 == 0) { // 如果nums1比较小,把nums2剩下的放入nums中
			while ((n2--) != 0) {
				nums[k] = nums2[j];
				j++;
				k++;
			}
		}else if (n2 == 0) { // 如果nums2比较小,把nums1剩下的放入nums中
			while ((n1--) != 0) {
				nums[k] = nums1[i];
				i++;
				k++;
			}
		}
		if ((nums1.length + nums2.length) % 2 == 0) { // 如果数字个数为偶数
			return (nums[(nums1.length + nums2.length) / 2] + nums[(nums1.length + nums2.length) / 2 - 1]) /2.0;
		} else { // 如果数字个数为奇数
			return nums[(nums1.length + nums2.length) / 2];
		}

	}

	public static void main(String[] args) {
		int[] nums1 = {};
		int[] nums2 = { 2, 3 };
		A4两个排序数组的中位数 a4两个排序数组的中位数 = new A4两个排序数组的中位数();
		System.out.println(a4两个排序数组的中位数.findMedianSortedArrays(nums1, nums2));
	}
}

需要注意的地方如下所示:

if (n1 == 0) { // 如果nums1比较小,把nums2剩下的放入nums

while ((n2--) != 0) {

nums[k] = nums2[j];

j++;

k++;

}

} else if (n2 == 0) { // 如果nums2比较小,把nums1剩下的放入nums

while ((n1--) != 0) {

nums[k] = nums1[i];

i++;

k++;

}

}

*如果不加else,上面那个判断n2--后,n2变成0,又会满足下面那个判断的条件。


return (nums[(nums1.length + nums2.length) / 2] + nums[(nums1.length + nums2.length) / 2 - 1]) / 2.0;

*减1是因为下标从0开始。

*除以2.0是为了让结果不要取整。


阅读更多
文章标签: LeetCode 中位数
个人分类: LeetCode
上一篇吴军的谷歌方法论|答读者问05|留学生怎样更好地融入美国?
下一篇LeetCode|5.最长回文子串(Java)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭