leetcode 4. Median of Two Sorted Arrays

寻找2个排序数组的中位数

也就是求第K大的数字

分割法：

k1=k2=K/2-1，即都分成一半(这里的k1 k2用于下标，故和数量相关的话要加1)

若分割下标比数组数量大，则取分割下标取数组最后一位

若nums1[k1]<nums2[k2]

那么数组1的前k1个可以舍去，接着求第K-k1-1大的数字

同理

若相等，看一下k1+k2+2是否刚好等于Kth，若相等，那么肯定就是这个数字

否则都剔除掉，继续求

int getKth(int l1,int l2,int Kth,vector<int> nums1, vector<int> nums2){int k,k1,k2;printf("l1=%d,l2=%d,Kth=%d\n",l1,l2,Kth);if(l1>=nums1.size()) //若nums1被剔除光了，则直接返回nums2的中位数return nums2[l2+Kth-1];if(l2>=nums2.size()) //若nums2被剔除光了，则直接返回nums1的中位数return nums1[l1+Kth-1];if(Kth==1)  //若要找第一大的数字，返回最小的那个即可return nums1[l1]<nums2[l2]?nums1[l1]:nums2[l2];k1=k2=Kth/2-1; //找两个分割点k1,k2，刚开始默认为Kth/2-1                //k1和k2用于分割点下标，即[l1+k1]，故不是第几大的意思if(k1>=nums1.size()) //若下标k1大于等于数组总数，则下标置为最大下标k1=nums1.size()-1;if(k2>=nums2.size()) //同理k2=nums2.size()-1;if(nums1[l1+k1]<nums2[l2+k2])  //剔除1数组前k1个return getKth(l1+k1+1,l2,Kth-k1-1,nums1,nums2);else if(nums1[l1+k1]>nums2[l2+k2]) //剔除2数组前k2个return getKth(l1,l2+k2+1,Kth-k2-1,nums1,nums2);else{if(k1+1+k2+1==Kth)  //下标总和加2才是数量总数。若数量综述等于Kth，说明这2个相等的就是要找的return nums1[l1+k1];else return getKth(l1+k1+1,l2+k2+1,Kth-k1-k2-2,nums1,nums2); //否则就都剔除，然后找后面的}}double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {int n1,n2,k1,k2;double num1,num2;n1=nums1.size();n2=nums2.size();if((n1+n2)%2==1){k1=(n1+n2)/2+1;num1=getKth(0,0,k1,nums1,nums2);return num1;}else {k1=(n1+n2)/2;num1=getKth(0,0,k1,nums1,nums2);k2=(n1+n2)/2+1;num2=getKth(0,0,k2,nums1,nums2);return (num1+num2)/2;}    }