【LeetCode】4. Median of Two Sorted Arrays

最新推荐文章于 2022-05-30 21:09:51 发布

永远的EMT

最新推荐文章于 2022-05-30 21:09:51 发布

阅读量265

点赞数

分类专栏：二分法 LeetCode 线性表

本文链接：https://blog.csdn.net/KID_LWC/article/details/76100457

版权

LeetCode 同时被 3 个专栏收录

85 篇文章 0 订阅

订阅专栏

线性表

18 篇文章 0 订阅

订阅专栏

二分法

6 篇文章 0 订阅

订阅专栏

题解：二分法，本题一般情况是查找第k大的数，由于是有序数组可以借鉴二分查找的思想，每次查找从k/2开始筛选

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
     int len1=nums1.size();
     int len2=nums2.size();
     int len= len1+len2;
     if(len%2){
        return findKth(nums1,0,nums2,0,len/2+1);

     }
     return (findKth(nums1,0,nums2,0,len/2)+findKth(nums1,0,nums2,0,len/2+1))/2;
}
double findKth(vector<int>& nums1,int n1, vector<int>& nums2,int n2,int k) {
    if(n1>=nums1.size()) return nums2[k+n2-1];
    if(n2>=nums2.size()) return nums1[k+n1-1];
    if(k==1) return min(nums1[n1],nums2[n2]);
    int ky1=n1+k/2-1>=nums1.size()?INT_MAX:nums1[n1+k/2-1];
    int ky2=n2+k/2-1>=nums2.size()?INT_MAX:nums2[n2+k/2-1];
    if(ky1<ky2) return findKth(nums1,n1+k/2,nums2,n2,k-k/2);
    else return findKth(nums1,n1,nums2,n2+k/2,k-k/2);   

}