leetcode-4

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

思路：求中位数转化为求topk，难点在于如何2个数组中求出topk。topk思路为每次比较为k/2，当k==1时，就是所求答案

int findTopk(vector<int>& nums1, vector<int>& nums2, int k) {
    //cout << "top:" << k << endl;
    int nums1_len = nums1.size();
    int nums2_len = nums2.size();
    if (nums1_len == 0 && nums2_len == 0) {
        return 0;
    }
    if (nums1_len == 0) {
        return nums2[k - 1];
    }
    if (nums2_len == 0) {
        return nums1[k - 1];
    }

    int nums1_idx = 0;
    int nums2_idx = 0;
    int nums1_new_idx = 0;
    int nums2_new_idx = 0;
    while (k != 1) {
        int idx = k / 2 - 1;

       //获取每个数组中的下标
        nums1_new_idx = min(nums1_idx + idx, nums1_len - 1);
        nums2_new_idx = min(nums2_len - 1, nums2_idx + idx);
        if (nums1[nums1_new_idx] > nums2[nums2_new_idx]) {
            k = k - (nums2_new_idx - nums2_idx + 1);
            nums2_idx = nums2_new_idx + 1;
        } else {
            k = k - (nums1_new_idx - nums1_idx + 1);
            nums1_idx = nums1_new_idx + 1;
        }

        //针对数据边界的问题加的条件，k+nums2_idx获取当前topk中的k，-1为数组中的下标
        if (nums1_idx == nums1_len) {
            return nums2[k + nums2_idx - 1];
        }
        if (nums2_idx == nums2_len) {
            return nums1[k + nums1_idx - 1];
        }
        //cout << "loop " <<  k << " " << nums1_idx << " " << nums2_idx << endl;
    }
    //cout << nums1_idx << " " << nums2_idx << " " << min(nums1[nums1_idx], nums2[nums2_idx]) << endl;
    return min(nums1[nums1_idx], nums2[nums2_idx]);
}

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
    int nums1_len = nums1.size();
    int nums2_len = nums2.size();
    if ((nums1_len + nums2_len) % 2 == 1) {
        return findTopk(nums1, nums2, (nums1_len + nums2_len) / 2.0 + 1);
    } else {
        return (findTopk(nums1, nums2, (nums1_len + nums2_len) / 2) + findTopk(nums1, nums2, (nums1_len + nums2_len) / 2 + 1)) / 2.0;
    }
}