Median of Two Sorted Arrays

package leetcode.xuyi;

import org.junit.Test;
/**
* Created by ethan on 2015/5/4.
*/
public class MedianOfTwoSortedArrays {
// 思路1： 由于时间复杂度限制在O(log(m+n))，只能使用二分法
// 其实这道题可以一般化到求最小的第n位数
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len = nums1.length + nums2.length;
if ((len&0x1)==1)
return findMedian(nums1, 0, nums2, 0, len/2+1);
else
return (findMedian(nums1, 0, nums2, 0, len/2)+findMedian(nums1, 0, nums2, 0, len/2+1))/2.0;
}
public double findMedian(int[] nums1, int start1, int[] nums2, int start2,  int target){
// 判断参数
int len1 = nums1.length - start1;
int len2 = nums2.length - start2;
if (len1 > len2)
// return 关键字不能少啊。。。 调了一个小时啊，妈蛋
return findMedian(nums2, start2, nums1, start1, target);
if (len1 == 0) {
return nums2[start2 + target - 1];
}
if (target == 1){
return  Math.min(nums1[start1], nums2[start2]);
}
int pa = Math.min(target/2, len1);
int pb = target - pa;
if(nums1[start1+pa-1] < nums2[start2+pb-1]){
return findMedian(nums1, start1+pa, nums2, start2, target-pa);
}else  if (nums1[start1+pa-1] > nums2[start2+pb-1]){
return findMedian(nums1, start1, nums2, start2+pb, target-pb);
}else
return nums1[start1+pa-1];
}

// 思路2： 先排序,O(m+n) 显然不能满足O(log(m+n)的要求) 但是能Accepted
public double findMedianSortedArrays1(int[] nums1, int[] nums2) {
if (nums1.length == 0 && nums2.length==0)
return 0;
int cur_1 = 0;
int cur_2 = 0;
int[] ans = new int[nums1.length+nums2.length];
int index = 0;
while(cur_1< nums1.length && cur_2<nums2.length){
if (nums1[cur_1] < nums2[cur_2]){
ans[index++] = nums1[cur_1];
cur_1++;
}else {
ans[index++] = nums2[cur_2];
cur_2++;
}
}
while (cur_1< nums1.length){
ans[index++] = nums1[cur_1];
cur_1++;
}
while (cur_2< nums2.length){
ans[index++] = nums2[cur_2];
cur_2++;
}
int len = ans.length;
if (len%2 == 0)
return (ans[len/2] + ans[len/2-1])/2.0;
else
return ans[len/2];
}

@Test
public void test4(){
System.out.println(findMedianSortedArrays(new int[]{2}, new int[]{}));
}
}