LeetCode - The World's Leading Online Programming Learning Platform
题目大意
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
解题思路
一、最容易想到的办法是把两个数组合并,然后取出中位数。但是合并有序数组的操作是 O(m+n) 的,不符合题意。看到题目给的 log 的时间复杂度,很容易联想到二分搜索。
class Solution {public double findMedianSortedArrays(int[] nums1, int[] nums2) {int size = nums1.length + nums2.length;int index1 = 0;int index2 = 0;int end = size/2;int begin = end;if (size % 2 == 0) {begin = begin-1;}int [] result = new int[end];int index = 0;while (index1 + index2 < size && index <= end) {if (index1 > nums1.length - 1) {result[index++] = nums2[index2++];continue;}if (index2 > nums2.length - 1) {result[index++] = nums1[index1++];continue;}if (nums1[index1] <= nums2[index2]) {result[index++] = nums1[index1++];continue;}result[index++] = nums2[index2++];}return (result[begin] + result[end])/2d;}
}
二、去中位进行两个数组移动