1029. Median (25)

最新推荐文章于 2019-02-07 23:51:33 发布

原创最新推荐文章于 2019-02-07 23:51:33 发布 · 2.4k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ZJU #PAT

面试编程训练同时被 2 个专栏收录

93 篇文章

订阅专栏

浙大PAT

54 篇文章

订阅专栏

本文介绍了一种通过合并两个已排序序列来找到合并后序列中位数的方法。使用C++实现，通过逐个比较两个序列中的元素并选择较小者进行推进，直到找到位于中间位置的元素。

考察合并序列的中位数

#include<vector>
#include<iostream>

int main()
{
	int n1, n2;
	while(scanf("%d",&n1)!=EOF)
	{
		std::vector<int> a, b;
		for(int i = 0; i < n1; ++i)
		{
			int tmp;
			scanf("%d",&tmp);
			a.push_back(tmp);
		}
		scanf("%d",&n2);
		for(int i = 0; i < n2; ++i)
		{
			int tmp;
			scanf("%d",&tmp);
			b.push_back(tmp);
		}
		//find median
		int aIdx, bIdx, k;
		aIdx = bIdx = 0;
		int mIdx = (n1+n2-1)/2;
		int cur;
		for( k = -1; aIdx < n1 && bIdx < n2; )
		{
			if(a[aIdx] <= b[bIdx]) 
			{
				cur = a[aIdx];
				++aIdx;
				++k;
			}
			else 
			{
				cur = b[bIdx];
				++bIdx;
				++k;
			}
			if(k == mIdx) 
			{
				printf("%d\n",cur);
				break;
			}

		}
		if(k != mIdx && aIdx < n1)
			printf("%d\n", a[aIdx+mIdx-k-1]);
		if(k != mIdx && bIdx < n2)
			printf("%d\n", b[bIdx+mIdx-k-1]);
	}
	return 0;
}