(水)
题目大意
给出两个非递减的数列,找出总数列的中位数。如果合并后数组长度为偶数则输出中间两个数中左边的那个。
思路
双指针。 注意为了防止数组越界,手动在a和b数组最后一个元素后加上一个很大的数INF。
样例测试数据:
2 1 2 7 3 4 5 6 7 8 9
AC代码
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<vector>
#define INF 10000000000
using namespace std;
int main() {
vector<long int>a, b;
int n1, n2;
int i;
long int temp;
scanf("%d", &n1);
for (i = 0;i < n1;i++) {
scanf("%ld", &temp);
a.push_back(temp);
}
a.push_back(INF);
scanf("%d", &n2);
for (i = 0;i < n2;i++) {
scanf("%ld", &temp);
b.push_back(temp);
}
b.push_back(INF);
int middle = (n1 + n2 + 1) / 2 - 1;
int po1 = 0, po2 = 0;
bool aorb = false;
i = 0;
while (i <= middle) {
if (a[po1] > b[po2]) {
po2++;
aorb = true;
}
else {
po1++;
aorb = false;
}
i++;
}
if (aorb) printf("%ld", b[po2 - 1]);
else printf("%ld", a[po1 - 1]);
return 0;
}