Update linkedlist_stack.cpp

Add test code.
This commit is contained in:
Yudong Jin 2022-11-28 21:08:10 +08:00 committed by GitHub
parent 06424ef023
commit c18affcea3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,36 +5,80 @@
*/ */
#include "../include/include.hpp" #include "../include/include.hpp"
#include<list>
/* 基于链表实现的栈 */ /* 基于链表实现的栈 */
class LinkedListStack { class LinkedListStack {
list<int> lst; private:
list<int> list;
public: public:
LinkedListStack() { LinkedListStack() {
// 初始化空链表 // 初始化空链表
lst.clear(); list.clear();
} }
/* 获取栈的长度 */ /* 获取栈的长度 */
int size() { int size() {
return lst.size(); return list.size();
} }
/* 判断栈是否为空 */ /* 判断栈是否为空 */
bool isEmpty() { bool empty() {
return lst.empty(); return list.empty();
} }
/* 入栈 */ /* 入栈 */
void push(int num) { void push(int num) {
lst.push_back(num); list.push_back(num);
} }
/* 出栈 */ /* 出栈 */
int pop() { int pop() {
int oldTop = lst.back(); int oldTop = list.back();
lst.pop_back(); list.pop_back();
return oldTop; return oldTop;
} }
/* 访问栈顶元素 */ /* 访问栈顶元素 */
int top() { int top() {
return lst.back(); return list.back();
}
/* 将 List 转化为 Array 并返回 */
vector<int> toVector() {
vector<int> vec;
for (int num : list) {
vec.push_back(num);
}
return vec;
} }
}; };
/* Driver Code */
int main() {
/* 初始化栈 */
LinkedListStack* stack = new LinkedListStack();
/* 元素入栈 */
stack->push(1);
stack->push(3);
stack->push(2);
stack->push(5);
stack->push(4);
cout << "栈 stack = ";
vector<int> vec = stack->toVector();
PrintUtil::printVector(vec);
/* 访问栈顶元素 */
int top = stack->top();
cout << "栈顶元素 top = " << top << endl;
/* 元素出栈 */
int pop = stack->pop();
cout << "出栈元素 pop = " << pop << ",出栈后 stack = ";
vec = stack->toVector();
PrintUtil::printVector(vec);
/* 获取栈的长度 */
int size = stack->size();
cout << "栈的长度 size = " << size << endl;
/* 判断是否为空 */
bool empty = stack->empty();
cout << "栈是否为空 = " << empty << endl;
}