NC36 在两个长度相等的排序数组中找到上中位数
描述
给定两个有序数组arr1和arr2,已知两个数组的长度都为N,求两个数组中所有数的上中位数。
上中位数:假设递增序列长度为n,若n为奇数,则上中位数为第n/2+1个数;否则为第n/2个数
[要求]
时间复杂度为O(logN)O(logN),额外空间复杂度为O(1)O(1)
示例1
输入:
[1,2,3,4],[3,4,5,6]
复制
返回值:
3
复制
说明:
总共有8个数,上中位数是第4小的数,所以返回3。
import java.util.*;
public class Solution {
/**
* find median in two sorted array
* @param arr1 int整型一维数组 the array1
* @param arr2 int整型一维数组 the array2
* @return int整型
*/
public int findMedianinTwoSortedAray (int[] arr1, int[] arr2) {
// write code here
if(arr1 == null && arr2 == null){
return 0 ;
}
if(arr1 == null){
if(arr2.length % 2 == 0){
return arr2[(arr2.length / 2)-1] ;
}else{
return arr2[(arr2.length / 2)+1] ;
}
}
if(arr2 == null){
if(arr1.length % 2 == 0){
return arr1[(arr1.length / 2)-1] ;
}else{
return arr2[arr1.length / 2] ;
}
}
int len = arr1.length + arr2.length ;
int n = 0 ;
if(len % 2 == 0){
n = (len / 2) -1 ;
}else{
n = len / 2 ;
}
int i = 0 , j = 0 , l =0 ;
int[] num = new int[n+1];
while(i < arr1.length && j < arr2.length){
if(arr1[i] < arr2[j]){
num[l] = arr1[i] ;
i++ ;
}else{
num[l] = arr2[j] ;
j++ ;
}
if(l == n){
return num[l];
}
l++ ;
}
while(i < arr1.length){
num[l] = arr1[i];
if(l == n){
return num[l];
}
i++ ;
l++ ;
}
while(j < arr2.length){
num[l] = arr2[j];
if(l == n){
return num[l];
}
j++ ;
l++ ;
}
return 0 ;
}
}