1029. Median (25)

1.这道题目与leetcode中的Median of Two Sorted Arrays相似，只是偶数情况取前一个，不用求平均

2.采用二分法进行查找

数组分为如下部分

{a[0],a[1],a[2],....a[i-1] | a[i],a[i+1],...a[m-1]}

{b[0],b[1],b[2],....b[j-1] | b[j],b[j+1],...b[n-1]}

然后根据a[i-1]与b[j]、a[i]与b[j-1]的大小，进行二分查找

3.注意边界条件，可以看代码的注释

//#include<string>
//#include <iomanip>
#include<vector>
#include <algorithm>
//#include<stack>
#include<set>
#include<queue>
#include<map>
//#include<unordered_set>
//#include<unordered_map>
//#include <sstream>
//#include "func.h"
//#include <list>
#include<stdio.h>
#include<iostream>
#include<string>
#include<memory.h>
#include<limits.h>
using namespace std;
/*
0
0

1 100
0

0
1 100

1 10
1 100

1 10
2 20 30

2 10 20
2 30 40

3 10 20 30
3 40 50 60

4 10 20 30 70
4 40 50 60 80

3 10 20 30
4 40 50 60 80

2 3 4
0

*/

/*
{a[0],a[1],a[2],....a[i-1]  |   a[i],a[i+1],...a[m-1]}
{b[0],b[1],b[2],....b[j-1]  |   b[j],b[j+1],...b[n-1]}

*/
int findMedianSortedArrays(int *nums1, int*nums2,int m,int n) {

if (m>n) return findMedianSortedArrays(nums2, nums1,n,m);
int minIdx = 0, maxIdx = m;//第一个数组的范围
int i, j;//两个数组的下标
int num1, num2;
int mid = (m + n + 1) >> 1;//为什么+1
while (minIdx <= maxIdx)
{
i = (minIdx + maxIdx) >> 1;//取中间值
j = mid - i;
if (i<m && j>0 && nums2[j - 1] > nums1[i])
minIdx = i + 1;
else if (i>0 && j<n && nums2[j] < nums1[i - 1])
maxIdx = i - 1;
else
{
if (i == 0) num1 = nums2[j - 1];
else if (j == 0) num1 = nums1[i - 1];
else num1 = max(nums1[i - 1], nums2[j - 1]);
break;
}
}
return num1;
}

int main(void)
{
int m, n;
scanf("%d", &m);
int *nums1 = new int[m];
for (int i = 0; i < m; i++)
{
scanf("%d", &nums1[i]);
}
scanf("%d", &n);
int *nums2 = new int
;
for (int i = 0; i < n; i++)
{
scanf("%d", &nums2[i]);
}
//cout << findMedianSortedArrays(nums1, nums2, m, n) << endl;

int *a, *b;

if (m > n)
{
a = nums2;
b = nums1;
swap(m, n);
}
else
{
a = nums1;
b = nums2;
}
/*
{a[0],a[1],a[2],....a[i-1]  |   a[i],a[i+1],...a[m-1]}
{b[0],b[1],b[2],....b[j-1]  |   b[j],b[j+1],...b[n-1]}
*/
int ans = INT_MIN;
bool getAns = false;
int l = 0, r = m;//必须是r=m，数组a是长度较小的一个，假如长度为0，那么m-1=-1，下面会直接出错
int i = (l + r) / 2;
int j = (m + n + 1) / 2 - i;
while (l <= r)//同理，假如m==0，至少确保循环执行了一次
{
i = (l + r) / 2;//假如m为0，i为0
j = (m + n + 1) / 2 - i;//+1很重要
if (i > 0 && j<n && a[i - 1]>b[j])
r = i - 1;
else if (j > 0 && i<m && b[j - 1]>a[i])
l = i + 1;
else
{
if (i == 0)
ans = b[j - 1];
else if (j == 0)
ans = a[i - 1];
else
ans = max(a[i - 1], b[j - 1]);
break;
}
}
cout << ans << endl;

return 0;
}