LeedCode刷题笔记-寻找两个正序数列中位数

LeedCode刷题笔记-寻找两个正序数列中位数

题目描述

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2
示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5
示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000
示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000
示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

解题代码

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
    //分别记录总长度中位数的位置index，还有两个数组的下标n1_index,n2_index
    int index = 0, n1_index = 0, n2_index = 0, k = 0;
    //记录当前位置的值cur_val,前一个位置的值pre_val
    double pre_val = 0, cur_val = 0;
    index = (nums1Size + nums2Size) / 2;
    
    //对两个数组进行遍历，按照大小顺序进行，一共执行index次，index为总长度中位数的位置
    for(k = 0; k <= index; k++)
    {
        if(n1_index < nums1Size && n2_index < nums2Size)
        {
            if(nums1[n1_index] > nums2[n2_index])
            {
                pre_val = cur_val;
                cur_val = nums2[n2_index];
                n2_index++;
                continue;
            }
            else
            {
                pre_val = cur_val;
                cur_val = nums1[n1_index];
                n1_index++;
                continue;
            }
        }
        //当其中一个数组全部遍历完成，仍未遍历index次数时，再对剩下一个数组进行遍历，index为两个数组中位数位置
        if(n2_index < nums2Size)
        {
            pre_val = cur_val;
            cur_val = nums2[n2_index];
            n2_index++;
            continue;
        }
         //当其中一个数组全部遍历完成，仍未遍历index次数时，再对剩下一个数组进行遍历，index为两个数组中位数位置
        if(n1_index < nums1Size)
        {
            pre_val = cur_val;
            cur_val = nums1[n1_index];
            n1_index++;
            continue;
        }
    }
    
    //最后针对不同的两个数组总长度输出中位数
    if((nums1Size + nums2Size) % 2)
    {
        //总长度和2取余，剩下1，所以总长度为奇数，直接返回当前值cur_val
        return cur_val;
    }
    else
    {
        //总长度和2取余，剩下0，所以总长度为奇数，需会两数的一半
        return (cur_val + pre_val) / 2;
    }
}

总结

• 可能会有长度为0的数组，可能一个数组遍历完成依然未到达中位数位置，所以必须是用for循环记录遍历次数
• continue是结束本次循环剩余代码，重新开始下一次循环，break是直接跳出循环体
• 自己刚开始的想法，没有考虑到单个数组遍历完成依然未取到中位数、0数组这些情况，自己是直接把所有判断代码放入了while(n1_index < nums1Size && n2_index < nums2Size)中
• 所以要考虑特殊情况，用特殊情况去思考一下自己的代码逻辑，是否是可以正确执行