LeetCode: Median of Two Sorted Arrays

kcnoize

于 2013-09-28 21:50:12 发布

阅读量808

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/kcnoize/article/details/12144421

本文介绍了一个C++实现的算法，用于找出两个已排序数组中第k大的元素及其应用场景——寻找中位数。该算法通过二分搜索策略高效地解决了问题。

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

查找第k大的数

class Solution {

public:
double findKthElem(int A[],int m,int B[],int n,int k){
int alow,amid,ahi,blow,bmid,bhi;
alow = 0,ahi = m-1,blow = 0,bhi = n-1;
while(alow<=ahi&&blow<=bhi){
amid = alow+(ahi-alow)/2;
bmid = blow+(bhi-blow)/2;
if(A[amid]<B[bmid]){
if(k<=amid-alow+1+bmid-blow)
bhi = bmid-1;
else{
k -= amid-alow+1; //k先减
alow = amid+1;
}
}
else{
if(k<=amid-alow+1+bmid-blow)
ahi = amid-1;
else{
k -= bmid-blow+1;
blow = bmid+1;
}
}
}
if(alow>ahi){
return B[blow+k-1];
}
if(blow>bhi)
return A[alow+k-1];
}

double findMedianSortedArrays(int A[], int m, int B[], int n) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
if((m+n)%2==1)
return findKthElem(A,m,B,n,(m+n+1)/2);
else
return (findKthElem(A,m,B,n,(m+n)/2)+findKthElem(A,m,B,n,(m+n)/2+1))/2;
}
};