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

一个更普通的问题：寻找第k个数。
记两个数组为A,B。
比较mid1 = A[k/2]和mid2 = B[k/2]，如果前者比较小，那说明A[k/2]肯定小于要找的第k个数。如果数组B的size小于k/2，那相当于mid2 = INT_MAX。接下去就从剩下的A和全部的B中找第k-k/2个数。

特殊情况：
k=1，只要比较两个数组的第一个数；
有一个数组为空，只要从另一个数组中取第k个数；
以上两种情况结束递归。

#include<limits.h>
class Solution {
public:
    int findkth(vector<int>& nums1, int st1, vector<int>& nums2, int st2, int k){
        if(st1>=nums1.size()) return nums2[st2+k-1];
        if(st2>=nums2.size()) return nums1[st1+k-1];
        if(k==1) return min(nums1[st1],nums2[st2]);
        int mid1 = st1+(k>>1)-1<nums1.size()?nums1[st1+(k>>1)-1]:INT_MAX;
        int mid2 = st2+(k>>1)-1<nums2.size()?nums2[st2+(k>>1)-1]:INT_MAX;
        if(mid1<mid2) return findkth(nums1, st1+(k>>1),nums2,st2,k-(k>>1));
        else return findkth(nums1, st1,nums2,st2+(k>>1),k-(k>>1));

    }
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        double ans=0;
        int a,b;
        a = b = (nums1.size()+nums2.size())/2;
        if(!((nums1.size()+nums2.size())%2)) --a;
        ans += findkth(nums1,0,nums2,0,a+1)+findkth(nums1,0,nums2,0,b+1);
        ans/=2;
        return ans;
    }
};