在Acwing上学习数据结构的一点笔记
栈:遵循“后进先出”的原则。即最后入栈的元素将首先被弹出。
用数组模拟栈:模拟栈包括两个基本操作插入和弹出,体现在代码上就是,每次有元素要入栈时,指向栈顶的指针会随之后移,指向新加入的元素。接下来是演示如何用数组模拟栈。
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int tt = -1, m; //tt表示栈顶索引,初始化为-1
int stack[N];
void push(int x)
{
stack[++ tt] = x; //栈顶所在索引往后移一格,放入x
}
void pop()
{
-- tt; //弹出栈顶元素,tt往前移
}
bool isEmpty() //创建bool数组,判断栈是否为空
{
if (tt == -1) return true;
else return false;
}
int query()
{
return stack[tt]; //返回栈顶元素
}
int main()
{
int x;
cin >> m;
string op;
while (m--) {
cin >> op;
if (op == "push")
{
cin >> x;
push(x);
}
else if (op == "query")
{
cout<<query()<<endl;
}
else if (op == "pop")
{
pop();
}
else if (op == "empty")
{
if (isEmpty())
{
cout << "YES";
}
else
{
cout << "NO";
}
cout << endl;
}
}
return 0;
}
单调栈
实现思路:创建一个严格单调递增的栈。即要对每一次对输入的数字进行判断,如果要加入的数字x要小于等于此时的栈顶元素,即stk[tt] >= x,则该栈顶元素不可能是答案,要将其弹出。而如果要加入的数字x要大于此时的栈顶元素,即stk[tt] < x,那么栈顶元素就是离它最近的比它小的值,直接输出stk[tt]。
#include <iostream>
using namespace std;
const int N = 100010;
int stk[N], tt, n;
int main()
{
cin>>n;
while (n --)
{
int x;
cin>>x;
while(tt && stk[tt] >= x) tt --; //如果栈不会空,且加入的数小于等于栈顶元素,则栈顶元素出栈。
if(!tt) cout<<-1<<' '; //如果栈为空,即没有答案,输出-1
else cout<<stk[tt]<<' ';
stk[++tt] = x; //将栈顶索引前移,并放入x
}
}