PAT1029 Median

本文提供了一个解决两个整数数组合并并找到中位数问题的手动排序算法示例,该算法避免了使用标准库排序带来的性能问题。

用标准库的排序全部超时,需要自己实现,另外还不能用cin cout

 

 

Sample Input

4 11 12 13 14
5 9 10 15 16 17
Sample Output
13

 

 

 

 

 

 

 

#include <stdio.h>

int N1[1000000];
int N2[1000000];
int M[2000000];


int main()
{
	int lenN1;
	scanf("%d",&lenN1);
	for (int i = 0; i < lenN1; i++)
	{
		scanf("%d",&N1[i]);
	}

	int lenN2;
	scanf("%d",&lenN2);
	for (int i = 0; i < lenN2; i++)
	{
		scanf("%d",&N2[i]);
	}

	int i = 0, j = 0, k = 0;
	while(i < lenN1 && j < lenN2)
	{
		if (N1[i] < N2[j])
		{
			M[k++] = N1[i++];
		}
		else if (N1[i] > N2[j])
		{
			M[k++] = N2[j++];
		}
		else
		{
			M[k++] = N1[i++];
			M[k++] = N2[j++];
		}
	}

	if (i < lenN1 && j >= lenN2)
	{
		while(i < lenN1)
		{
			M[k++] = N1[i++];
		}
	}
	else
	{
		while(j < lenN2)
		{
			M[k++] = N2[j++];
		}
	}

	if ((lenN1 + lenN2)%2)
	{
		printf("%d",M[(lenN1+lenN2)/2]);
	}
	else
	{
		printf("%d",M[(lenN1+lenN2)/2 - 1]);
	}

	return 0;
}
 
### PAT 甲级 真题 1057 解析 #### 题目描述 题目要求实现一个特殊的栈结构,除了基本的 `Push` 和 `Pop` 操作外,还需要支持一个新的操作 `PeekMedian`。该操作返回当前栈中所有元素的中位数值。对于偶数个元素的情况,定义中位数为第 \( \frac{N}{2} \) 小的元素;对于奇数个元素,则取第 \( \left(\frac{N+1}{2}\right) \) 小的元素[^2]。 #### 实现方法一:基于树状数组的方法 为了高效处理大量的查询请求而不至于超时,可以采用树状数组来追踪栈内的元素分布情况,并通过二分法定位所需的中位数位置。这种方法的时间复杂度大约为 \(O(n\log n)\),能够满足性能需求[^1]。 ```cpp // 示例代码片段展示如何利用树状数组解决此问题 #include <iostream> using namespace std; const int MAX_N = 1e5 + 5; int c[MAX_N], a[MAX_N]; void update(int i, int delta){ while(i<MAX_N){ c[i]+=delta; i+=i&(-i); } } int sum(int i){ int s=0; while(i>0){ s+=c[i]; i-=i&(-i); } return s; } ``` #### 实现方法二:使用 multiset 维护中位数指针的方式 另一种解决方案涉及到了 C++ STL 中的容器——`multiset`。借助于这个有序集合的数据特性以及额外设置的一个指向中位数位置的迭代器,在执行插入或删除动作的同时调整迭代器的位置即可快速获取最新的中位数。 ```cpp // 使用 multiset 的C++实现方式 #include <set> typedef long long LL; #define PB push_back #define MP make_pair #define FI first #define SE second #define PII pair<int,int> class MedianFinder { public: /** initialize your data structure here. */ set<pair<int ,LL>> S; // 存储数据及其版本号 set<pair<int ,LL>>::iterator it; // 记录中位数所在位置 void addNum(int num) { static LL ver = 0; ++ver; auto p = MP(num, ver); if(S.size() == 0 || S.size()%2==0){ S.insert(p),it=S.find(p); }else{ bool flag=false; if(it->FI<num){ if(next(it)!=S.end()){ if((next(it)->FI)>num){ --it,S.insert(p); flag=true; } } }else{ if(it!=S.begin()){ if(prev(it)->FI<=num){ ++it,S.insert(p); flag=true; } } } if(!flag)S.insert(p); } } double findMedian(){ int sz=(int)(S).size(); if(sz%2==0)return (*prev(it)).FI*0.5+(*it).FI*0.5; else return *it->FI; } }; ```
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值