问题描述
There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
思路分析
用到归并排序里面的关于两个有序数组合并的思想。
- 如果两数组总数是奇数,则中位数是最中间值,偶数则为最中间两个数均值;
- 可能会出现某一数组已找完还没有到达中位数,需要继续在另一个数组上向后搜寻。
代码
class Solution {
public:
double findMedianSortedArrays(int A[], int m, int B[], int n) {
double ans=0;
int len = (m + n + 1) / 2;
int flag = (m + n) % 2; //用来标记中位数是一个数还是两个数的均值(1代表一个数,0代表两数均值)
int finish = 0; //用来表示寻找工作是否结束
int count = 0;
int i=0,j=0;
while (i < m&&j < n)
{
if (A[i] < B[j])
{
count++;
if (count == len) {
ans += A[i]; if (flag) { finish = 1; break; }
}
if (count == len + 1){
ans += A[i]; finish = 1; break;
}
i++;
}
else {
count++;
if (count == len) {
ans += B[j]; if (flag){ finish = 1; break; }
}
if (count == len + 1){
ans += B[j]; finish = 1; break;
}
j++;
}
}
while (!finish&&i < m)
{
count++;
if (count == len) { ans += A[i]; if (flag) break; }
if (count == len + 1){ ans += A[i]; break; }
i++;
}
while (!finish&&j < n)
{
count++;
if (count == len) { ans += B[j]; if (flag) break; }
if (count == len + 1){ ans += B[j]; break; }
j++;
}
ans = flag ? ans : ans / 2;
return ans;
}
};