7-5 特殊堆栈(二分+树状数组/线段树)

已于 2023-03-28 22:13:47 修改
阅读量252 收藏
点赞数 1
分类专栏： 天梯赛训练 文章标签： 算法 数据结构 c++
于 2023-03-28 22:12:11 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_60008503/article/details/129826127
版权

传送门

1. 首先看题意很容易就得出暴力解法, 即出/入栈正常模拟, 找中值就通过对当前栈内的数进行排序, 再输出中值即可, 复杂度O(n^n^logn)。

代码如下:

#include <bits/stdc++.h>

using namespace std;

#define endl "\n"
#define debug cout<<"debug"<<endl
mt19937 rd(time(0));
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
const double eps = 1e-8;
const double PI = 3.14159265358979323;
const int N = 2e5+10, M = 2*N, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n;
stack<int> stk;

void solve()
{
    cin>>n;
    vector<int> v;
    while( n -- )
    {
        string s; cin>>s;
        if(s == "Pop")
        {
            if(stk.size())
            {
                cout<<stk.top()<<endl; stk.pop();
                v.erase(v.end()-1);
            }else cout<<"Invalid"<<endl;
        }else if(s == "PeekMedian")
        {
            if(stk.size())
            {
                vector<int> tmp = v;
                sort(tmp.begin(), tmp.end());
                cout<<tmp[(tmp.size()-1)/2]<<endl;
            }else cout<<"Invalid"<<endl;
        }else{
            int x; cin>>x;
            stk.push(x);
            v.push_back(x);
        }
    }
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    int T;
    T = 1;
//    cin>>T;
    while(T -- )
    {
        solve();
    }
    return 0;
}

很明显, 这样的做法铁T, 但是能骗到17分, 还可以

2. 优化, 考虑一个cnt数组, cnt[x]表示栈内x出现的次数, 对从1开始的cnt数组, 依次求和, 当和sum>=中位即(n+1)/2时, 当前位就是中值, 时间复杂度 O(n^n)

#include <bits/stdc++.h>

using namespace std;

#define endl "\n"
#define debug cout<<"debug"<<endl
mt19937 rd(time(0));
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
const double eps = 1e-8;
const double PI = 3.14159265358979323;
const int N = 2e5+10, M = 2*N, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n;
stack<int> stk;
int cnt[N];

void solve()
{
    cin>>n;
    vector<int> v;
    while( n -- )
    {
        string s; cin>>s;
        if(s == "Pop")
        {
            if(stk.size())
            {
                cnt[stk.top()]--;
                cout<<stk.top()<<endl; stk.pop();
            }else cout<<"Invalid"<<endl;
        }else if(s == "PeekMedian")
        {
            if(stk.size())
            {
                ll sum = 0;
                for(int i = 1; i<=1e5; i++)
                {
                    sum += cnt[i];
                    if(sum>=(stk.size()+1)/2)
                    {
                        cout<<i<<endl;
                        break;
                    }
                }
            }else cout<<"Invalid"<<endl;
        }else{
            int x; cin>>x;
            stk.push(x);
            cnt[x]++;
        }
    }
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    int T;
    T = 1;
//    cin>>T;
    while(T -- )
    {
        solve();
    }
    return 0;
}

漂亮, 又多骗了5分hhh

3. 最终优化, 考虑二分, 我们要找的是一个数, 这个数和前面所有数的出现次数之和, 来判断这个次数和是不是到了中位, 这个过程是可以二分的, 但问题是如何快速的获取前面所有数出现的次数之和, 我一开始想的是树状数组, 但是脑子抽了不太会写, 选择用线段树的解法, 线段树维护区间数字出现的次数和, 每次入栈就对应区间+1反之-1, 查询中值时, 就通过查询线段树来快速的获取前面所有数出现的次数和, 时间复杂度O(n^logn^logn)

代码如下:

#include <bits/stdc++.h>

using namespace std;

#define endl "\n"
#define debug cout<<"debug"<<endl
mt19937 rd(time(0));
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
const double eps = 1e-8;
const double PI = 3.14159265358979323;
const int N = 2e5+10, M = 2*N, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n;
stack<int> stk;
int cnt[N];

struct Node{
    int l, r, sum;
}tr[N<<2];

void pushup(int u)
{
    tr[u].sum = tr[u<<1].sum + tr[u<<1|1].sum;
}

void build(int u, int l, int r)
{
    tr[u] = {l, r};
    if(l == r) return;
    int mid = l + r >> 1;
    build(u<<1, l, mid), build(u<<1|1, mid+1, r);
}

int query(int u, int l, int r)
{
    if(tr[u].l>=l && tr[u].r<=r) return tr[u].sum;
    int mid = tr[u].l + tr[u].r >> 1;
    int res = 0;
    if(l<=mid) res = query(u<<1, l, r);
    if(r>mid) res += query(u<<1|1, l, r);
    return res;
}

void modify(int u, int x, int k)
{
    if(tr[u].l == x && tr[u].r == x)
    {
        if(k == 1) tr[u].sum++;
        else tr[u].sum = max(0, tr[u].sum-1);
        return;
    }
    int mid = tr[u].l + tr[u].r >> 1;
    if(x<=mid) modify(u<<1, x, k);
    else modify(u<<1|1, x, k);
    pushup(u);
}

void solve()
{
    cin>>n;
    vector<int> v;
    build(1, 1, 1e5);
    while( n -- )
    {
        string s; cin>>s;
        if(s == "Pop")
        {
            if(stk.size())
            {
                modify(1, stk.top(), 0);
                cout<<stk.top()<<endl; stk.pop();
            }else cout<<"Invalid"<<endl;
        }else if(s == "PeekMedian")
        {
            if(stk.size())
            {
                int l = 1, r = 1e5;
                int x = (stk.size()+1)/2;
                while(l<r)
                {
                    int mid = l+r >> 1;
                    if(query(1, 1, mid)>=x) r = mid;
                    else l = mid + 1;
                }
                cout<<l<<endl;
            }else cout<<"Invalid"<<endl;
        }else{
            int x; cin>>x;
            stk.push(x);
            modify(1, x, 1);
        }
    }
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    int T;
    T = 1;
//    cin>>T;
    while(T -- )
    {
        solve();
    }
    return 0;
}

然后就ac啦, 但显然还有更好写的做法

下面给出一开始没想到的树状数组优化, 思维同线段树但实现起来更简单: 

#include <bits/stdc++.h>

using namespace std;

#define endl "\n"
#define debug cout<<"debug"<<endl
mt19937 rd(time(0));
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
const double eps = 1e-8;
const double PI = 3.14159265358979323;
const int N = 2e5+10, M = 2*N, mod = 1e9+7;
const int INF = 0x3f3f3f3f;

int n;
stack<int> stk;
int tr[N];

int lowbit(int x)
{
    return x & -x;
}

void add(int x, int c)
{
    for( ; x<=1e5; x += lowbit(x)) tr[x] += c;
}

int ask(int x)
{
    int res = 0;
    for( ; x; x -= lowbit(x)) res += tr[x];
    return res;
}

void solve()
{
    cin>>n;
    vector<int> v;
    while( n -- )
    {
        string s; cin>>s;
        if(s == "Pop")
        {
            if(stk.size())
            {
                add(stk.top(), -1);
                cout<<stk.top()<<endl; stk.pop();
            }else cout<<"Invalid"<<endl;
        }else if(s == "PeekMedian")
        {
            if(stk.size())
            {
                int l = 1, r = 1e5;
                int x = (stk.size()+1)/2;
                while(l<r)
                {
                    int mid = l+r >> 1;
                    if(ask(mid)>=x) r = mid;
                    else l = mid + 1;
                }
                cout<<l<<endl;
            }else cout<<"Invalid"<<endl;
        }else{
            int x; cin>>x;
            stk.push(x);
            add(x, 1);
        }
    }
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    int T;
    T = 1;
//    cin>>T;
    while(T -- )
    {
        solve();
    }
    return 0;
}

樱落二瓣七里香
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
特殊堆栈
一个小渣渣的博客：）
01-16 1102
这题其实就是甲级题目的A1057 Stack，题目的大意是让你模拟操作堆栈的入栈出栈，同时能够求出栈中元素的中值，这个中值的意思是指讲栈中所有元素排完序后，这一串有序元素的中值。 如果通过常规的排序来操作的话，这题基本就是超时，无法AC,30分的题也只能拿17分左右。于是我们需要用一些更巧妙的方法来做这道题。我学习了柳婼大神的方法，使用树状数组来解决这道题。 树状数组的讲解我这边因为也不是太会画...
特殊堆栈 (30 分)
weixin_45954447的博客
04-15 489
输入样例： Pop PeekMedian Push 3 PeekMedian Push 2 PeekMedian Push 1 PeekMedian Pop Pop Push 5 Push 4 PeekMedian Pop Pop Pop Pop 输出样例： Invalid Invalid 3 2 2 1 2 4 4 5 3 Invalid 题目意思是找中位数，当然和中位数还有点不一样。 首先我们必定要有个栈，但是我们不能光有一个栈。这样不能找已有元素的中值。所以我设置了一个vector，理由如下 ve
#训练5# 4-特殊堆栈
syzyaaa
04-08 150
题目 取中值，是中位数，而不是中间的那个数啊啊啊。。。 代码 #include <iostream> #include <iostream> #include <cstring> #include <stack> #include <algorithm> #include <queue> #include <vector> #include <string> #define MAXX 10010 using
基础实验3-2.5 堆栈模拟队列（stack）
木静音yy的博客
02-28 287
题目 设已知有两个堆栈S1和S2，请用这两个堆栈模拟出一个队列Q。 所谓用堆栈模拟队列，实际上就是通过调用堆栈的下列操作函数: int IsFull(Stack S)：判断堆栈S是否已满，返回1或0； int IsEmpty (Stack S )：判断堆栈S是否为空，返回1或0； void Push(Stack S, ElementType item )：将元素item压入堆栈S； ElementType Pop(Stack S )：删除并返回S的栈顶元素。 实现队列的操作，即入队void AddQ(Ele
数据结构与算法——栈的实现及模拟
m0_62179366的博客
05-13 520
目录 一、栈的原理 二、栈的实现 1.栈的定义 2.栈的初始化 3.入栈 4.出栈 5.获取栈顶元素 6.栈的大小 7.判断栈是否为空 8.栈的销毁 一、栈的原理 堆栈（英语：stack）又称为栈或堆叠，是计算机科学中的一种抽象资料类型，只允许在有序的线性资料集合的一端（称为堆栈顶端，英语：top）进行加入数据（英语：push）和移除数据（英语：pop）的运算。因而按照后进先出（LIFO, Last In First Out）的原理运作。 常与另一种有序的线性资料集合
Object-C-使用Object-C实现的堆栈stacks+队列queues.zip
最新发布
03-03
Object-C_使用Object-C实现的堆栈stacks+队列queues
kt-micronaut-grpc-DEMO:堆栈kotlin + micronaut + grpc的演示应用
04-10
kt-micronaut-grpc-DEMO 堆栈kotlin + micronaut + grpc的演示应用
mern-boilerplate：React + Express + Webpack + Mongo = MERN堆栈样板
02-05
样板-MongoDB，Express，React，NodeJS状态依存关系Python 2（node-gyp） NodeJS（9岁以上） Docker（CE很好，首选17+）产品特点全栈javascript，因此获取最新版本的Node！ Webpack 4 React 表达 萨斯 ES6-Stage-0 +...
28kb-react-redux-routing, 响应+ Redux + 路由堆栈仅 28kb.zip
10-10
28kb-react-redux-routing, 响应+ Redux + 路由堆栈仅 28kb + Redux + Routing + 路由堆栈，用于 kb只是因为你的堆栈很棒并不意味着它必须是胖的。这个 repo 作为一个最小可以行的例子，如何结合反应+ on的大概念而...
stacks-Queues_DSA:C ++的堆栈和队列
04-03
stacks-Queues_DSA C ++的堆栈和队列
模拟栈与队列
Tom的博客
06-29 218
手写数据结构中的栈与队列，模拟单调栈和单调队列。
堆中的路径【堆：插入+到根结点的路径】
sankuaizhuobu的博客
08-12 150
练习4.3堆中的路径(25分) 将一系列给定数字插入一个初始为空的小顶堆H[]。随后对任意给定的下标i，打印从H[i]到根结点的路径。 输入格式: 每组测试第1行包含2个正整数N和M(≤1000)，分别是插入元素的个数、以及需要打印的路径条数。下一行给出区间[-10000, 10000]内的N个要被插入一个初始为空的小顶堆的整数。最后一行给出M个下标。 输出格式: 对输入中给出的每个下标i，在一行中输出从H[i]到根结点的路径上的数据。数字间以1个空格分隔，行末不得有多余空格。 输入样例: ..
团体天梯 L3-002 特殊堆栈 （30 分）
ItaLink
03-04 1624
L3-002特殊堆栈（30 分） 堆栈是一种经典的后进先出的线性结构，相关的操作主要有“入栈”（在堆栈顶插入一个元素）和“出栈”（将栈顶元素返回并从堆栈中删除）。本题要求你实现另一个附加的操作：“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素，如果 N 是偶数，则中值定义为第 N/2 小元；若是奇数，则为第 (N+1)/2 小元。 输入格式： 输入的第一行是正整数 N（≤1...
L3-002 特殊堆栈 (30 分)
姚军
11-17 540
题目链接 知识点： lower_bound（begin, end, value）大于等于value的迭代器位置 版本1 有序向量 #include <iostream> #include <string> #include <algorithm> #include <stack> #include <vector> using names...
【PTA】特殊堆栈
阳光大男孩！！！的博客
01-31 5483
题目重述 堆栈是一种经典的后进先出的线性结构，相关的操作主要有“入栈”（在堆栈顶插入一个元素）和“出栈”（将栈顶元素返回并从堆栈中删除）。本题要求你实现另一个附加的操作：“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素，如果 N 是偶数，则中值定义为第 N/2 小元；若是奇数，则为第 (N+1)/2 小元。 输入格式： 输入的第一行是正整数 N（≤105​​ ）。随后 N 行，每行给...
pta7-5 出栈序列的合法性 (25 分)
qq_37071435的博客
05-21 2176
给定一个最大容量为M的堆栈，将N个数字按 1, 2, 3, ...,N的顺序入栈，允许按任何顺序出栈，则哪些数字序列是不可能得到的？例如给定M=5、N=7，则我们有可能得到{ 1, 2, 3, 4, 5, 6, 7 }，但不可能得到{ 3, 2, 1, 7, 5, 6, 4 }。 输入格式： 输入第一行给出 3 个不超过 1000 的正整数：M（堆栈最大容量）、N（入栈元素个数）...
7-13 特殊堆栈 (30 分)
Foolishman的博客
04-23 1237
  堆栈是一种经典的后进先出的线性结构，相关的操作主要有“入栈”（在堆栈顶插入一个元素）和“出栈”（将栈顶元素返回并从堆栈中删除）。本题要求你实现另一个附加的操作：“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素，如果 N 是偶数，则中值定义为第 N/2 小元；若是奇数，则为第 (N+1)/2 小元。 输入格式：   输入的第一行是正整数 N（≤10​5​​ ）。随后 N 行，每行给出一句指令，为以下 3 种之一： Push key Pop PeekMedian   其中 key 是不超过
pta L3-002 特殊堆栈 (模拟堆栈输出中值）
暖心的博客
03-02 1140
题目链接 思路： 用stack模拟输入输出，用vector每次按顺序插入值，用lower_bound()函数 #include &lt;bits/stdc++.h&gt; using namespace std; typedef long long ll; const int maxn = 1e4+10; const int INF=0x3f3f3f3f; #define fir ...
pta7-5 特殊堆栈c语言
06-07
好的，我理解了你的问题。那么，pta7-5 特殊堆栈的 C 语言实现如下： ```c #include <stdio.h> #include <stdlib.h> #define MAX_SIZE 100 // 定义特殊堆栈结构体 struct SpecialStack { int data[MAX_SIZE]; // 堆栈数据 int min[MAX_SIZE]; // 堆栈中的最小值 int top; // 堆栈顶部指针 }; // 初始化特殊堆栈 void initStack(struct SpecialStack *stack) { stack->top = -1; } // 判断特殊堆栈是否为空 int isEmpty(struct SpecialStack *stack) { return stack->top == -1; } // 判断特殊堆栈是否已满 int isFull(struct SpecialStack *stack) { return stack->top == MAX_SIZE - 1; } // 向特殊堆栈中添加元素 void push(struct SpecialStack *stack, int item) { if (isFull(stack)) { printf("特殊堆栈已满，无法添加元素！\n"); return; } // 添加元素到堆栈中 stack->data[++stack->top] = item; // 更新最小值 if (stack->top == 0) { stack->min[stack->top] = item; } else { int min = stack->min[stack->top - 1]; stack->min[stack->top] = item < min ? item : min; } } // 从特殊堆栈中弹出元素 int pop(struct SpecialStack *stack) { if (isEmpty(stack)) { printf("特殊堆栈为空，无法弹出元素！\n"); return -1; } // 取出堆栈顶部元素 int item = stack->data[stack->top--]; // 更新最小值 stack->min[stack->top + 1] = 0; return item; } // 获取特殊堆栈中的最小值 int getMin(struct SpecialStack *stack) { if (isEmpty(stack)) { printf("特殊堆栈为空，无法获取最小值！\n"); return -1; } return stack->min[stack->top]; } // 测试函数 int main() { // 创建特殊堆栈 struct SpecialStack stack; initStack(&stack); // 向特殊堆栈中添加元素 push(&stack, 3); push(&stack, 5); push(&stack, 2); push(&stack, 1); // 获取特殊堆栈中的最小值 printf("特殊堆栈中的最小值为：%d\n", getMin(&stack)); // 弹出特殊堆栈中的元素 printf("特殊堆栈中的元素依次为："); while (!isEmpty(&stack)) { printf("%d ", pop(&stack)); } printf("\n"); return 0; } ``` 以上是特殊堆栈的 C 语言实现代码，你可以根据需要进行修改、调试和运行。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值