PAT A 1029 Median (25 分)

最新推荐文章于 2022-04-06 15:50:57 发布

Siumai

最新推荐文章于 2022-04-06 15:50:57 发布

阅读量363

点赞数

分类专栏： PAT A 文章标签： PAT PAT A

本文链接：https://blog.csdn.net/qq_43749739/article/details/98996172

版权

PAT A 专栏收录该内容

140 篇文章 1 订阅

订阅专栏

一、思路

每次比较使用ans保存两元素中的较小元素，比较 (N1 + N2 + 1)/ 2次获得第 (N1 + N2 + 1)/ 2小的元素即Median元素。

二、代码

#include <cstdio>
#include <vector>
using namespace std;
int main()
{
	int N[2], ans = 0;
	vector<int> a[2];
	for( int i = 0; i < 2; ++i )
	{
		scanf("%d", &N[i]);
		a[i].resize(N[i]);
		for( int j = 0; j < N[i]; ++j )
			scanf("%d", &a[i][j]);
	}
	for( int i = 0, j = 0, k = 0; k < (N[0] + N[1] + 1) / 2; ++k )
		if( (i < N[0] && j >= N[1]) || (i < N[0] && j < N[1] && a[0][i] <= a[1][j]))
			ans = a[0][i++];
		else ans = a[1][j++];
	printf("%d", ans);
}