pat 甲级A1029 Median (25分)

本文介绍了一种高效算法,用于合并两个已排序的整数序列,并找到合并后序列的中位数。该算法在输入第二个序列的同时进行比较和计数,以确定中位数的位置。

题目链接:

https://pintia.cn/problem-sets/994805342720868352/problems/994805466364755968

 

题目分析:

给出两个递增序列,将两者合并后的中位数输出。本题有点类似多项式的加法,可在输入第二个递增序列时,同时得到最终的结果。

 

参考代码:

#include <iostream>
using namespace std;
int s1[200001];
int main() {
    int n, m;
    cin >> n;
    for (int i = 0; i < n; i++)  scanf("%d", &s1[i]);
    s1[n] =  1e9;
    cin >> m;
    int midPos = (n + m + 1) / 2;
    int i = 0, temp, count = 0;
    for (int j = 0; j < m; j++) {  //在输入第二个序列的时候,可进行与第一个序列的比较
        scanf("%d", &temp);
        for(;s1[i] < temp; i++) {   
            count++;
            if (midPos == count)  printf("%d", s1[i]);
        }
        count++;
        if (midPos == count)  printf("%d", temp);
    }
    for(; i < n; i++) {
        count++;
        if (midPos == count)  printf("%d", s1[i]);
    }
    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、付费专栏及课程。

余额充值