[LeetCode]Median of Two Sorted Arrays 二分查找两个有序数组的第k数(中位数)

本文介绍了一种利用二分查找法解决数组排序问题的高效算法,通过比较两个数组中对应位置的元素,逐步缩小搜索范围,最终确定目标元素的位置。详细解释了算法的实现过程,并提供了代码示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

二分。情况讨论

由于数组有序,所以可以考虑用二分。通过二分剔除掉肯定不是第k位数的区间。假设数组A和B当前处理的下标分别是mid1和mid2,则

1、如果A[mid1]<B[mid2],

①、若mid1+mid2+2==k(+2是因为下标是从0开始的),则

mid1在大有序数组中下标肯定小于k,所以可以排除[0,mid1]。此外,B[mid2]下标大于或等于k,可以排除[mid2+1,n];

②、若mid1+mid2+2<k,则

mid1在大有序数组中下标肯定小于k,所以可以排除[0,mid1]

③、若mid1+mid2+2>k,则

B[mid2]下标大于k,可以排除[mid2,n];

2、如果A[mid1]<B[mid2]情况相符,只是下标改变。


这些操作处理完后,可能一个数组被排除了,即满足lowX>highX,此时只需对另一个数组进行二分,同时二分其元素在另一个数组中的下标,确定全局下标,最终通过判断全局下标与k的关系,确定是否为第k数

class Solution {
public:
    int findPos(int* p,int n,int x){
        int low=0,high=n-1,mid;
        while(low<=high){
            mid=(low+high)>>1;
            if(p[mid]<=x)low=mid+1;
            else high=mid-1;
        }
        return low;
    }
	double findK(int a[], int m, int b[], int n,int k){
		int mid1,mid2,low1=0,low2=0,high1=m-1,high2=n-1,x;
        while(low1<=high1&&low2<=high2){
            mid1=(high1+low1)>>1;
            mid2=(high2+low2)>>1;
            if(a[mid1]<b[mid2]){
                if(mid1+mid2+2==k){
					low1=mid1+1;
					high2=mid2;
				}
                else if(mid1+mid2+2<k){
                    low1=mid1+1;
                }
                else high2=mid2-1;
            }
            else{
                if(mid1+mid2+2==k){
					low2=mid2+1;
					high1=mid1;
				}
                else if(mid1+mid2+2<k){ 
                    low2=mid2+1;
                }
                else high1=mid1-1;
            }
        }
        if(low1<=high1){
		//	if(low1==high1)return a[low1];
            while(low1<=high1){
                mid1=(low1+high1)>>1;
                x=findPos(b,n,a[mid1]);
                if(x+mid1+1==k)return a[mid1];
                else if(x+mid1<k)low1=mid1+1;
                else high1=mid1-1;
            }
            return low1>=m?a[m-1]:a[low1];
        }
        else {
	//		if(low2==high2)return b[low2];
            while(low2<=high2){
                mid2=(low2+high2)>>1;
                x=findPos(a,m,b[mid2]);
                if(x+mid2+1==k)return b[mid2];
                else if(x+mid2<k)low2=mid2+1;
                else high2=mid2-1;
            }
            return low2>=n?a[n-1]:b[low2];
        }
	}
    double findMedianSortedArrays(int a[], int m, int b[], int n) {
		int k=m+n;
        if(k&1){
			return findK(a,m,b,n,k/2+1);
		}
		else{
			return (findK(a,m,b,n,k/2)+findK(a,m,b,n,k/2+1))/2.0;
		}
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值