[PAT] A1029 Median

tech2022-09-06  102

(水)

题目大意

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

思路

双指针。 注意为了防止数组越界,手动在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; }
最新回复(0)