leetcode-4-寻找两个有序数组的中位数

 

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int N = nums1.size(), M = nums2.size(), cnt=0, n1=0, n2=0;
        double res = 0;
        bool ave = !((M + N) % 2), flag = false;
        while (n1 < N && cnt<(M+N)/2+1) { //n1未完
            if ( n2 < M) { //n2未完
                while (nums1[n1] <= nums2[n2]) {
                    n1++;
                    cnt++;
                    if (ave && cnt == (M + N) / 2) {
                        res = nums1[n1 - 1];
                        flag = true;
                    }
                    if (ave && cnt == (M + N) / 2+1) return (res + nums1[n1 - 1]) / 2.0;
                    if (!ave && cnt == (M + N) / 2+1) return nums1[n1 - 1];
                    if (n1 >= N) break;
                }
                if (n1 >= N) break;
                while (nums1[n1] > nums2[n2]) {
                    n2++;
                    cnt++;
                    if (ave && cnt == (M + N) / 2) {
                        res = nums2[n2 - 1];
                        flag = true;
                    }
                    if (ave && cnt == (M + N) / 2+1) return (res + nums2[n2 - 1]) / 2.0;
                    if (!ave && cnt == (M + N) / 2+1) return nums2[n2 - 1];
                    if (n2 >= M) break;
                }
            }
            else { //n2已完
                if (res != 0) return (res + nums1[n1]) / 2.0;
                else if (ave) return (nums1[n1 + (M + N) / 2 - cnt] + nums1[n1 + (M + N) / 2 - cnt - 1]) / 2.0;
                else if (!ave) return nums1[n1 + (M + N) / 2 - cnt];
            }
        }
        if (flag) return (res + nums2[n2]) / 2.0;
        else if (ave) return (nums2[n2 + (M + N) / 2 - cnt] + nums2[n2 + (M + N) / 2 - cnt - 1]) / 2.0;
        else if (!ave) return (nums2[n2 + (M + N) / 2 - cnt]);
        return 0;
    }
};

int main() {
    Solution sol;
    vector<int> input1 = {1,4};
    vector<int> input2 = {2,8};
    cout << sol.findMedianSortedArrays(input1, input2);
    cout << endl;
}