数组-leetcode#4-两个升序数组的中位数

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
       if((m+n)%2==1)  
           return find_k(nums1,nums2,(m+n+1)/2);
       else 
           return (find_k(nums1,nums2,(m+n)/2)+find_k(nums1,nums2,(m+n)/2+1))/2.0;
    }
    double find_k(vector<int>& nums1, vector<int>& nums2,int k){
        int m = nums1.size();
        int n = nums2.size();
        int i=0,j=0;
        int count = 0;
        while(i<m && j<n){
            int min_ele = min(nums1[i],nums2[j]);
            count++;
            if(k==count) return min_ele;
            if(nums1[i]==min_ele) i++;
            else j++;
        }
        if(i<m) return nums1[i+(k-count)-1];
        else return nums2[j+(k-count)-1];
    }
};

有点意外，一把过了，效果还不错[捂脸]。第一想法就是对于奇数个，找到两个数组中第(m+n+1)/2个数，对于偶数个，找到两个数组中第(m+n)/2和第(m+n)/2+1这两个数，求均值。写一个找两个数组中第k个数字的函数。