4.寻找两个正序数组的中位数_java_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

题解

按顺序依次取两个数组的值，直到取到中间值即可
至于时间复杂度我就不会整了

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
		//两数组总长
        int len = nums1.length + nums2.length;
		//中间值的索引
		int midindex = len / 2;
		//定义两个变量进行迭代，总长为奇数，中位数就是mid2；总长为偶数，中位数就是mid1和mid2的平均值
		int mid1 = 0,mid2 = 0;
		//定义两个指针分别作为两个数组的索引
		int i1 = 0, i2 = 0;
		while (true) {
			if (i1 + i2 == midindex + 1) break;
			//依次迭代
			mid1 = mid2;
			//当某个指针已经到达边界，只迭代另一个指针
			if (i1 == nums1.length) { mid2 = nums2[i2++]; continue; }
			if (i2 == nums2.length) { mid2 = nums1[i1++]; continue; }
			//两指针都为到达边界，比较大小按顺序取值
			mid2 = nums1[i1] < nums2[i2] ? nums1[i1++] : nums2[i2++];
		}
		return len % 2 == 0? (mid1 + mid2) / 2.0 : mid2;
    }
}

作者：kf_ye-you-shen
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/an-shun-xu-yi-ci-qu-liang-ge-shu-zu-de-zhi-zhi-dao/
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。