题目
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/
1.二分查找
分析
对两个数组进行分割,二分查找分割点。
对每个数组而言,若数组长度是奇数,左边元素比右边元素多一个;若数组长度是偶数,分割后左右元素相等。
数组1的分割点记为i-1,i。
数组2的分割点记为j-1,j。
因为数组有序,所以nums1[i-1]<nums1[i],但数组间还要满足nums1[i-1]<nums2[j],nums2[j-1]<nums1[i]。
代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
//默认nums1是更短的
//如果nums1更长,交换num1和nums2。
if (nums1.length > nums2.length) {
return findMedianSortedArrays(nums2, nums1);
}
int m = nums1.length;
int n = nums2.length;
int left = 0, right = m;
// median1:前一部分的最大值
// median2:后一部分的最小值
int median1 = 0, median2 = 0;
while (left <= right) {
// 前一部分包含 nums1[0 .. i-1] 和 nums2[0 .. j-1]
// 后一部分包含 nums1[i .. m-1] 和 nums2[j .. n-1]
int i = (left + right) / 2;//数组1的分割点
int j = (m + n + 1) / 2 - i;//数组2的分割点
//对数组越界的情况巧妙地处理
int nums1a = (i == 0 ? Integer.MIN_VALUE : nums1[i - 1]);
int nums1b = (i == m ? Integer.MAX_VALUE : nums1[i]);
int nums2a = (j == 0 ? Integer.MIN_VALUE : nums2[j - 1]);
int nums2b = (j == n ? Integer.MAX_VALUE : nums2[j]);
if (nums1a <= nums2b) {
median1 = Math.max(nums1a, nums2a);//各自左半边的最大值
median2 = Math.min(nums1b, nums2b);//各自右半边的最小值
left = i + 1;
}
else {
right = i - 1;
}
}
return (m + n) % 2 == 0 ? (median1 + median2) / 2.0 : median1;
}
}
复杂度
结果
2.我的暴力解法
代码
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m=nums1.length;
int n=nums2.length;
if(m==0&&n==0)
return 0;
if(m==0){
if(n%2==0)
return (nums2[n/2-1]+nums2[n/2])/2.0;
else
return nums2[n/2];
}
if(n==0){
if(m%2==0)
return (nums1[m/2-1]+nums1[m/2])/2.0;
else
return nums1[m/2];
}
int mn=m+n;
int count=0;//count=mn/2 或
int i=0,j=0;
int a=0,b=0;
while(i<m&&j<n){
count++;
if(nums1[i]<=nums2[j]){
if(mn%2==1&&count==mn/2+1){
return nums1[i];
}
if(mn%2==0&&count==mn/2){
a=nums1[i];
}
if(mn%2==0&&count==mn/2+1){
b=nums1[i];
return (a+b)/2.0;
}
i++;
}
else{
if(mn%2==1&&count==mn/2+1){
return nums2[j];
}
if(mn%2==0&&count==mn/2){
a=nums2[j];
}
if(mn%2==0&&count==mn/2+1){
b=nums2[j];
return (a+b)/2.0;
}
j++;
}
}
while(i<m){
count++;
if(mn%2==1&&count==mn/2+1){
return nums1[i];
}
if(mn%2==0&&count==mn/2){
a=nums1[i];
}
if(mn%2==0&&count==mn/2+1){
b=nums1[i];
return (a+b)/2.0;
}
i++;
}
while(j<n){
count++;
if(mn%2==1&&count==mn/2+1){
return nums2[j];
}
if(mn%2==0&&count==mn/2){
a=nums2[j];
}
if(mn%2==0&&count==mn/2+1){
b=nums2[j];
return (a+b)/2.0;
}
j++;
}
return 0;
}
}
复杂度
时间复杂度 O(m+n):m,n是num1,nums2的长度
空间复杂度O(1)