[PAT] A1029 Median

（水）

题目大意

给出两个非递减的数列，找出总数列的中位数。如果合并后数组长度为偶数则输出中间两个数中左边的那个。

思路

双指针。
注意为了防止数组越界，手动在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;
}