题目及测试
package pid004;
/*寻找两个有序数组的中位数
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
*/
public class main {
public static void main(String[] args) {
int[][] testTable = {
{1, 3},{2},{1, 2},{3, 4}};
test(testTable[0],testTable[1]);
test(testTable[2],testTable[3]);
}
private static void test(int[] ito1,int[] ito2) {
Solution solution = new Solution();
double rtn;
long begin = System.currentTimeMillis();
for (int i = 0; i < ito1.length; i++) {
System.out.print(ito1[i]+" ");
}//开始时打印数组
System.out.println();
for (int i = 0; i < ito2.length; i++) {
System.out.print(ito2[i]+" ");
}//开始时打印数组
rtn = solution.findMedianSortedArrays(ito1,ito2);//执行程序
long end = System.currentTimeMillis();
System.out.println("rtn=" +rtn);
System.out.println();
System.out.println("耗时:" + (end - begin) + "ms");
System.out.println("-------------------");
}
}
解法1(成功,3ms,极快)
写一个循环,然后里边判断是否到了中位数的位置,到了就返回结果,但这里对偶数和奇数的分类会很麻烦。当其中一个数组遍历完后,出了 for 循环对边界的判断也会分几种情况。总体来说,虽然复杂度不影响,但代码会看起来很乱。
用 len 表示合并后数组的长度,如果是奇数,我们需要知道第 (len+1)/2 个数就可以了,如果遍历的话需要遍历 int(len/2 ) + 1 次。如果是偶数,我们需要知道第 len/2和 len/2+1 个数,也是需要遍历 len/2+1 次。所以遍历的话,奇数和偶数都是 len/2+1 次。
返回中位数的话,奇数需要最后一次遍历的结果就可以了,偶数需要最后一次和上一次遍历的结果。所以我们用两个变量 left 和 right,right 保存当前循环的结果,在每次循环前将 right 的值赋给 left。这样在最后一次循环的时候,left 将得到 right 的值,也就是上一次循环的结果,接下来 right 更新为最后一次的结果。
packa