PTA--特殊堆栈

堆栈是一种经典的后进先出的线性结构,相关的操作主要有“入栈”(在堆栈顶插入一个元素)和“出栈”(将栈顶元素返回并从堆栈中删除)。本题要求你实现另一个附加的操作:“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素,如果 N 是偶数,则中值定义为第 N/2 小元;若是奇数,则为第 (N+1)/2 小元。

输入格式:
输入的第一行是正整数 N(≤10
​5
​​ )。随后 N 行,每行给出一句指令,为以下 3 种之一:

Push key
Pop
PeekMedian
其中 key 是不超过 10
​5
​​ 的正整数;Push 表示“入栈”;Pop 表示“出栈”;PeekMedian 表示“取中值”。

输出格式:
对每个 Push 操作,将 key 插入堆栈,无需输出;对每个 Pop 或 PeekMedian 操作,在一行中输出相应的返回值。若操作非法,则对应输出 Invalid。

输入样例:
17
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

lower_bound:

lower_bound( begin,end,num):从数组的begin位置到end-1位置二分查找第一个大于或等于num的数字,找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。

解析

用一个vector 和 stack 实现,vector是作为有个有序的数组,用lower_bound 查找到插入位置后插入,stack就是作为堆栈,这样可以既实现了堆栈push和pop的功能,又能从vector中得到中值

题中的压入,弹出功能和求中值功能是分开的,因为不存在一个有序的堆栈,所以借助vector实现

lower_bound 利用二分查找法找到第一个大于或者等于查找数的数的位置

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int n, temp;
	scanf("%d", &n);
	vector<int> ve;
	stack<int> st;
	for (int i = 0; i < n; ++i)
	{
		char s[11];
		scanf("%s", s);
		if (s[1] == 'o'){
			if (st.empty()){
				cout << "Invalid" << endl;
			}else{
//因为是从小到大插入的
				auto it = lower_bound(ve.begin(), ve.end(), st.top());
				cout << *it << endl;
				ve.erase(it);
				st.pop();
			}
		}else if (s[1] == 'e'){
			if (st.empty()){
				cout << "Invalid" << endl;
			}else{
				if (st.size() % 2)cout << ve[ve.size() / 2] << endl;
				else cout << ve[ve.size() / 2 - 1] << endl;
			}
		}else{
			scanf("%d", &temp);
			st.push(temp);
			ve.insert(lower_bound(ve.begin(), ve.end(), st.top()), temp);
		}
	}
	return 0;
}

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我理解了你的问题。那么,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 语言实现代码,你可以根据需要进行修改、调试和运行。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值