题目大意:
给出两个已经递增的序列s1,s2,长度分别为N和M,求将它们合并成一个新的递增序列后的中位数(个数为偶数时为左半部分的最后一个数)
思路:
1)在给定两个子序列的长度N和M之后,新序列的长度就是已知的,中位数的位置为(N+M-1)/2向下取整。
2)定义计数器count,在对两个有序序列进行合并的时候统计已经合并的个数,直到count增长到中位数的位置时停止,此时,s1[i]与s2[j]中较小的为中位数。
3)令s1 s2两数组的最后一个元素的后一位一个极大的数,保证在遍历的时候i与j不会越界。
AC代码:
//PAT_A 1029
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1000010;
const int INF = 0x7fffffff;
int s1[maxn], s2[maxn];
int main() {
int n, m;
(void)scanf("%d", &n);
for (int i = 0; i < n; i++) {
(void)scanf("%d", &s1[i]);
}
(void)scanf("%d", &m);
for (int i = 0; i < m; i++) {
(void)scanf("%d", &s2[i]);
}
s1[n] = s2[m] = INF;//两序列最后一个元素设为INF
int mid = (n + m - 1) / 2;//中位数位置
int i = 0, j = 0, count = 0;
while (count < mid) {
if (s1[i] < s2[j])i++;
else j++;
count++;
}
printf("%d\n", min(s1[i],s2[j]));
return 0;
}