题目:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
方法一:
1.中位数:偶数个中间两个和除以2,奇数个中位数为中间那个
2.先判断是否其中有一个数组为空,如果有直接返回中位数即可
3.将两个数组中的值重新排序到新的数组,返回新数组的中位数
class FindMedianSortedArrays {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1=nums1.length;
int len2=nums2.length;
if(len1==0){
if(len2%2==0){
return (nums2[len2/2-1]+nums2[len2/2])/2.0;
}else{
return nums2[len2/2];
}
}
if(len2==0){
if(len1%2==0){
return (nums1[len1/2-1]+nums1[len1/2])/2.0;
}else{
return nums1[len1/2];
}
}
int[] arr=new int[len1+len2];
int count=0;
int m=0;
int n=0;
while(count!=(len1+len2)){
if(m==len1){
while(n!=len2){
arr[count++]=nums2[n++];
}
break;
}
if(n==len2){
while(m!=len1){
arr[count++]=nums1[m++];
}
break;
}
if(nums1[m]>nums2[n]){
arr[count++]=nums2[n++];
}else{
arr[count++]=nums1[m++];
}
}
int len=len1+len2;
if((len)%2==0){
return (arr[len/2-1]+arr[len/2])/2.0;
}else{
return arr[len/2];
}
}
}
方法二:
不需要将两个数组真的合并,我们只需要找到中位数在哪里就可以了
1.用 len 表示合并后数组的长度,如果是奇数,我们需要知道第 (len+1)/2 个数就可以了,如果遍历的话需要遍历 int(len/2 ) + 1 次。如果是偶数,我们需要知道第 len/2和 len/2+1 个数,也是需要遍历 len/2+1 次。所以遍历的话,奇数和偶数都是 len/2+1 次。
2.用两个变量 left 和 right,right 保存当前循环的结果,在每次循环前将 right 的值赋给 left。这样在最后一次循环的时候,left 将得到 right 的值,也就是上一次循环的结果,接下来 right 更新为最后一次的结果。
3.用aflag标志nums1的下标,用bflag标志nums2的下标
class FindMedianSortedArrays1 {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1=nums1.length;
int len2=nums2.length;
int len=len1+len2;
int aflag=0;
int bflag=0;
int left=0;
int right=0;
for(int i=0;i<=len/2;i++){
left=right;
if(aflag<len1&&(bflag>=len2||nums2[bflag]>nums1[aflag])){
right=nums1[aflag++];
}else{
right=nums2[bflag++];
}
}
if((len&1)==0){
return (left+right)/2.0;
}else{
return right;
}
}
}