在两个长度相等的排序数组中找到上中位数
给定两个有序数组arr1和arr2,已知两个数组的长度都为N,求两个数组中所有数的上中位数。
上中位数:假设递增序列长度为n,若n为奇数,则上中位数为第n/2+1个数;否则为第n个数
[要求]
时间复杂度为O(logN),额外空间复杂度为O(1)
代码
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 -1;
return find(arr1, 0, arr1.length-1, arr2, 0, arr2.length-1);
}
public int find(int[] arr1, int l1, int r1, int[] arr2, int l2, int r2){
int mid1 = l1 + ( (r1-l1)>>1 );
int mid2 = l2 + ( (r2-l2)>>1 );
if(l1 >= r1) return Math.min(arr1[l1], arr2[l2]);
// 元素个数为奇数,则offset为0,为偶数则 offset 为 1
int offset = ( (r1-l1+1)&1 ) ^ 1;
if(arr1[mid1] < arr2[mid2]) return find(arr1, mid1+offset, r1, arr2, l2, mid2);
else if(arr1[mid1] > arr2[mid2]) return find(arr1, l1, mid1, arr2, mid2+offset, r2);
else return arr1[mid1];
}
}