描述:给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。算法的时间复杂度应该为 O(log (m+n)) 。
部分来自:LeetCode
解法1:暴力破解
思路:找到第len/2+1,或者len/2和len/2+1的数
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int length=nums1.length+nums2.length;
double middle=0;
int index=0;//总的数组的下标
int i=0,j=0;
int temp;//暂时存放当前的值
if (length%2==0)//偶数的情况,即要求array[length/2-1]和array[length/2]的平均数
{
boolean flag=false;
int num1=0,num2 = 0;
for (;i<nums1.length&&j<nums2.length;)
{
if (nums1[i]<=nums2[j]){
temp=nums1[i];
i++;
}
else {
temp=nums2[j];
j++;
}
if (index==length/2-1)num1=temp;
if (index==length/2) {
num2 = temp;
flag=true;//代表已经拿到
break;
}
index++;
}
if (flag==false) {
if (index - 1 == length / 2 - 1)//此时num1已经写入,但num2并没有
{
if (i == nums1.length)
num2 = nums2[j];
else num2 = nums1[i];
} else if (index - 1 < length / 2 - 1) {//对应2个都没有被写入的情况
if (i == nums1.length) {
num1 = nums2[j + length / 2 - 1 - index];
num2 = nums2[j + length / 2 - index];
} else {
num1 = nums1[i + length / 2 - 1 - index];
num2 = nums1[i + length / 2 - index];
}
}
}
middle=((double) num1+(double)num2)/2;
}
else {//奇数的情况,即要的是array[length/2]
boolean flag=false;//若为true,则已经获得middle
for (;i<nums1.length&&j<nums2.length;)
{
if (nums1[i]<=nums2[j]){
temp=nums1[i];
i++;
}
else {
temp=nums2[j];
j++;
}
if (index==length/2) {
middle=temp;
flag=true;
break;
}
index++;
}
if (flag==false)
{
if (i==nums1.length)
middle=nums2[length/2-index+j];
else middle=nums1[length/2-index+i];
}
}
return middle;
}
这里有个小问题,处理特殊情况时,要考虑情况是否重合。最好的办法是如果已经拿到所需的值,用一个flag屏蔽后面的步骤。
解法二:暴力破解
思路:只需要找到中位数,同解法一
public double findMedianSortedArrays(int[] A, int[] B) {
int m = A.length;
int n = B.length;
int len = m + n;
int left = -1, right = -1;
int aStart = 0, bStart = 0;
for (int i = 0; i <= len / 2; i++) {
left = right;
if (aStart < m && (bStart >= n || A[aStart] < B[bStart])) {//重点
right = A[aStart++];
} else {
right = B[bStart++];
}
}
if ((len & 1) == 0)
return (left + right) / 2.0;
else
return right;
}
重点看此解法中找中位数的方式,很简单,比解法一更好。