leetcode Day7 4 寻找两个正序数组的中位数

题目

思路：
情况一：两个数组一者为空
情况二：两个数组没有重叠部分，即一者所有元素都比另一者大或小
情况三：两个数组有重叠部分

前两种情况都很简单，最难处理的是情况三，其方法如下
【loc：元素在两个数组合并后的新数组中的位置】
1.取出更长的数组作为第一层循环，另一个作为第二层循环
【注意循环这里可以不用从头轮询第二层循环】
2.长数组元素大于短数组元素，得到该短数组元素的位置loc
3.长数组元素小于短数组元素，得到该长数组元素的位置loc
4.当找到位于中间位置的元素后停止轮询，程序退出
【对于偶数个元素，还需要记录loc的上一个位置pri】
5.对于个别情况【中位数还没找到，一个数组就轮询结束】
显然，这时本质上两个数组都已经排序完毕了，我们只用调出位序正确的元素即可

代码：O(log (m+n))

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    int mid = (nums1Size+nums2Size+1)/2;//中位数的位置
    bool nod = (nums1Size+nums2Size+1)%2;//需不需要/2
    //情况一
    if(!nums1Size)return nod?(nums2[mid]+nums2[mid-1])/2.0:nums2[mid-1];
    if(!nums2Size)return nod?(nums1[mid]+nums1[mid-1])/2.0:nums1[mid-1];
    int i,j,pri,loc;
    int *p;
    if(nums1Size>nums2Size){//使nums2总是最长的
        p = nums1;
        nums1 = nums2;
        nums2 = p;
        i = nums1Size;
        nums1Size = nums2Size;
        nums2Size = i;
    }
    //情况二
    if(nums1[0]>=nums2[nums2Size-1]){
        if(nums1Size==nums2Size)return (nums1[0]+nums2[nums2Size-1])/2.0;
        return nod?(nums2[mid]+nums2[mid-1])/2.0:nums2[mid-1];
    }
    if(nums2[0]>=nums1[nums1Size-1]){
        if(nums1Size==nums2Size)return (nums2[0]+nums1[nums1Size-1])/2.0;
        return nod?(nums2[mid-nums1Size]+nums2[mid-nums1Size-1])/2.0:nums2[mid-nums1Size-1];
    }
    //情况三：两个数组互相有重叠部分
    j=0;bool t=false;
    for(i=0;i<nums2Size;i++){//长的
        for(;j<nums1Size;j++){//短的
            if(i+j<mid+1){
                if(nums2[i]<=nums1[j]){
                    pri = loc;
                    loc = nums2[i];
                    break;
                }else{
                    pri = loc;
                    loc = nums1[j];
                }
            }else{
                t = true;
                break;
            }
        }
        if(t)break;
        else if(i+j<mid+1&&j==nums1Size){
            pri = loc;
            loc = nums2[i];
        }
    }
    return nod?(pri+loc)/2.0: pri ;
}

提交结果：