算法题目训练--二分

序言

记录自己的算法升级打怪之路，持续更新

Leetcode系列

Leetcode P4.寻找两个正序数组的中位数

题目：

​

思路：如果没有时间复杂度，可以直接合并然后算中位数，但是设定了时间复杂度，由时间复杂度可知用二分来做。但是此题二分难度极大，极难理解，参考题解力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {

//任何长度的数组的中位数 可以用(length+1)/2 + (length+2)/2除2得到
        int left = (nums1.size() + nums2.size() + 1) / 2;
        int right = (nums1.size() + nums2.size() + 2) / 2;
        return (findmid(nums1, nums2, 0, 0, left) + findmid(nums1, nums2, 0, 0, right)) / 2;
    }


    double findmid(vector<int>& arr1, vector<int>& arr2, int i, int j, int k) {
//如果i超过了数组num1的长度，剩下的k直接在num2里找
        if (i >= arr1.size()) {
            return arr2[j + k - 1];
        }
        if (j >= arr2.size()) {
            return arr1[i + k - 1];
        }

//如果k只剩下一个了，说明这就是答案，选择两个数组中最小的那个
        if (k == 1) { return min(arr1[i], arr2[j]); }


        int mid1 = 0;
        int mid2 = 0;
        if (i + k / 2 - 1 < arr1.size()) {
            mid1 = arr1[i + k / 2 - 1];
        }
        else {
            mid1 = INT_MAX;
        }
        if (j + k / 2 - 1 < arr2.size()) {
            mid2 = arr2[j + k / 2 - 1];
        }
        else {
            mid2 = INT_MAX;
        }

        if (mid1 < mid2) {
            return findmid(arr1, arr2, i + k / 2, j, k - k / 2);
        }
        else {
            return findmid(arr1, arr2, i, j + k / 2, k - k / 2);
        }
    }
};

提交结果：