-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMedian of Two Sorted Arrays.cpp
46 lines (45 loc) · 1.32 KB
/
Median of Two Sorted Arrays.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
class Solution {
public:
double findMedianSortedArrays(int A[], int m, int B[], int n) {
if (m>n){
return findMedianSortedArrays(B, n, A, m);
}
if (m < 2){
if (m == 0) return (B[(n-1)/2] + B[n/2])/2.0;
if (m == 1){
if (n & 1){
if (n == 1){
return (A[0] + B[0]) / 2.0;
}
int x = B[n/2-1], y = B[n/2+1], z = B[n/2];
if (A[0] < x) return (x + z)/2.0;
if (A[0] < y) return (A[0] + z)/2.0;
return (z + y)/2.0;
}else{
int x = B[(n-1)/2], y = B[n/2];
if (A[0] < x) return x;
if (A[0] > y) return y;
return A[0];
}
}
}
int xa = 0, ya = 0, xb = 0, yb = 0;
if(A[(m-1)/2] < B[(n-1)/2]){
xa = m/2;
}else{
xb = n/2;
}
if (A[m/2] > B[n/2]){
ya = m/2;
}else{
yb = n/2;
}
if (xa != ya){
xa = min(xa, yb);
xb = min(xb, ya);
ya = min(ya, xb);
yb = min(yb, xa);
}
return findMedianSortedArrays(A+xa,m-xa-ya, B+xb, n-xb-yb);
}
};