cpp-algo-cases/chapter_stack_and_queue/array_stack.cpp

94 lines
1.7 KiB
C++

/**
* File: array_stack.cpp
* Created Time: 2022-11-28
* Author: qualifier1024 (2539244001@qq.com)
*/
#include "../utils/common.hpp"
/* 基于数组实现的栈 */
class ArrayStack
{
private:
vector<int> stack;
public:
/* 获取栈的长度 */
int size()
{
return stack.size();
}
/* 判断栈是否为空 */
bool isEmpty()
{
return stack.size() == 0;
}
/* 入栈 */
void push(int num)
{
stack.push_back(num);
}
/* 出栈 */
int pop()
{
int num = top(); // 先获取栈顶元素并保存
stack.pop_back(); // 再删除栈顶元素
return num; // 返回栈顶元素 (用于反馈给用户出栈元素是什么)
}
/* 访问栈顶元素 */
int top()
{
if (isEmpty())
throw out_of_range("栈为空"); // 如果栈为空,抛出异常
return stack.back(); // 返回栈顶元素
}
/* 返回 Vector */
vector<int> toVector()
{
return stack;
}
};
/* Driver Code */
int main()
{
/* 初始化栈 */
ArrayStack *stack = new ArrayStack();
/* 元素入栈 */
stack->push(1);
stack->push(3);
stack->push(2);
stack->push(5);
stack->push(4);
cout << "栈 stack = ";
printVector(stack->toVector());
/* 访问栈顶元素 */
int top = stack->top();
cout << "栈顶元素 top = " << top << endl;
/* 元素出栈 */
top = stack->pop();
cout << "出栈元素 pop = " << top << ",出栈后 stack = ";
printVector(stack->toVector());
/* 获取栈的长度 */
int size = stack->size();
cout << "栈的长度 size = " << size << endl;
/* 判断是否为空 */
bool empty = stack->isEmpty();
cout << "栈是否为空 = " << empty << endl;
// 释放内存
delete stack;
return 0;
}