数据结构 - 数组模拟栈/队列

数据结构 - 数组模拟栈/队列

一、模拟栈：

例题：

实现一个栈，栈初始为空，支持四种操作：

(1) “push x” – 向栈顶插入一个数x；

(2) “pop” – 从栈顶弹出一个数；

(3) “empty” – 判断栈是否为空；

(4) “query” – 查询栈顶元素。

现在要对栈进行M个操作，其中的每个操作3和操作4都要输出相应的结果。

输入格式
第一行包含整数M，表示操作次数。

接下来M行，每行包含一个操作命令，操作命令为”push x”，”pop”，”empty”，”query”中的一种。

输出格式
对于每个”empty”和”query”操作都要输出一个查询结果，每个结果占一行。

其中，”empty”操作的查询结果为“YES”或“NO”，”query”操作的查询结果为一个整数，表示栈顶元素的值。

数据范围
1≤M≤100000,
1≤x≤10⁹
所有操作保证合法。

输入样例：
10
push 5
query
push 6
pop
query
pop
empty
push 4
query
empty
输出样例：
5
5
YES
4
NO

---

代码：

#include<iostream>
#include<string>

using namespace std;

const int N=1e6+10;

int top,stk[N],n;

int main()
{
    cin>>n;
    string op;
    int x;
    while(n--)
    {
        cin>>op;
        if(op=="push") {cin>>x; stk[++top]=x;}
        else if(op=="pop") --top;
        else if(op=="empty")
        {
            if(top==0) puts("YES");
            else puts("NO");
        }
        else if(op=="query") cout<<stk[top]<<endl;
    }
    return 0;
}

---

二、模拟队列

例题:

实现一个队列，队列初始为空，支持四种操作：

(1) “push x” – 向队尾插入一个数x；

(2) “pop” – 从队头弹出一个数；

(3) “empty” – 判断队列是否为空；

(4) “query” – 查询队头元素。

现在要对队列进行M个操作，其中的每个操作3和操作4都要输出相应的结果。

输入格式
第一行包含整数M，表示操作次数。

接下来M行，每行包含一个操作命令，操作命令为”push x”，”pop”，”empty”，”query”中的一种。

输出格式
对于每个”empty”和”query”操作都要输出一个查询结果，每个结果占一行。

其中，”empty”操作的查询结果为“YES”或“NO”，”query”操作的查询结果为一个整数，表示队头元素的值。

数据范围
1≤M≤100000,
1≤x≤10⁹,
所有操作保证合法。

输入样例：
10
push 6
empty
query
pop
empty
push 3
push 4
pop
query
push 6
输出样例：
NO
6
YES
4

---

代码：

#include<iostream>
#include<string>

using namespace std;

const int N=1e5+10;

int n,x,que[N],hh,tt=-1;
string op;

int main()
{
    cin>>n;
    while(n--)
    {
        cin>>op;
        if(op=="push") {cin>>x; que[++tt]=x;}
        else if(op=="pop") ++hh;
        else if(op=="empty") 
        {
            if(hh>tt) puts("YES");
            else puts("NO");
        }
        else if(op=="query") cout<<que[hh]<<endl;
    }
    
    return 0;
}