栈的模拟
首先,我们需要了解什么是栈,栈究竟实现了什么样的模型,有个大致的了解,就可以开始用数组进行栈的功能的实现了(当然,Java有Stack进行实现,也会放在后面)
什么是栈:
栈是一种线性表,主要实现了“先进后出”的模式。
栈的模式在生活中可以类比为串糖葫芦,这个例子可以帮助我们理解。第一个糖葫芦相当于栈底元素,放到最底下,然后后面的糖葫芦再放进来,最后一个糖葫芦相当于栈顶元素。我们在取糖葫芦的时候只能从顶的一端进行操作,再一个个取出来,越后串上的糖葫芦越先出去,这就是栈的先进后出。形象理解这个例子或者记住下面的图可以更好地理解栈。
需要什么:
1.stk[]数组存放数据;
2.栈顶下标tt;
详解:怎么模拟栈
1.插入操作
先让栈顶“指针”往前走,再赋值
2.弹出操作
3.判断栈是否为空
4.取栈顶元素
题目
解题完整代码(Java)
import java.util.Scanner;
public class Main {
static int N = 100010;
static int[] stk = new int[N];
static int tt;//tt表示栈顶下标
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int M = sc.nextInt();
while (M-- > 0){
String s = sc.next();
if(s.equals("push")){
int x = sc.nextInt();
push(x);
}else if(s.equals("pop")){
pop();
}else if(s.equals("empty")){
if(!empty()){
System.out.println("NO");
}else {
System.out.println("YES");
}
}else {
System.out.println(getTop());
}
}
}
//插入操作
public static void push(int x){
stk[++tt] = x;
}
//弹出操作
public static void pop(){
tt--;
}
//判断栈是否为空
public static boolean empty(){
if(tt > 0){
return false;
}else {
return true;
}
}
//栈顶元素
public static int getTop(){
return stk[tt];
}
}
Java-API:
单调栈
单调栈出题的一种常见题型:
该元素左 / 右边第一个比它本身大 / 小的元素(求出元素值或者元素下标)
思路:
如果求左边第一个比它自身小的元素(其它情况进行类比,只是判断条件不同):
思路1:
每一个数都往前往后进行循环比较,但是复杂度高,数据量大就容易超时。
思路2:
思考一下,在比较中有哪些数在特定情况下永远不会再进行比较了?
在当前数前面的数,并且比当前数要小的数。
为什么?
当前数在与前面的数相比在靠后的位置,只要当前数可以拿到,那么前面的数就没有被拿到的机会了。
当前这个数本身如果比栈顶小的话,由于更靠右,当前的栈顶元素就没有用了,之后不会再用到。
题目:
解题完整代码(Java)
import java.util.Scanner;
public class Main {
static int N = 100010;
static int[] stk = new int[N];//模拟栈
static int tt;//栈顶指针
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
for (int i = 0; i < n; i++) {
int x = sc.nextInt();
//当栈不为空时,只要栈顶元素比x大,之后就永远不会用到,把x弹出
while (tt != 0 && stk[tt] >= x){
tt--;
}
//当栈顶元素比x小时,如果栈不为空,则栈顶元素就是前一个比它小的元素
if(tt != 0){
System.out.print(stk[tt] + " ");
}else {
//栈为空,表示前面没有比x小的元素
System.out.print(-1+" ");
}
//栈顶指针前移,把x存入栈中
stk[++tt] = x;
}
}
}
题目(蓝桥-百亿富翁)
解题完整代码(Java)
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int N = sc.nextInt();
//a存放原数组
int[] a = new int[N+1];
for (int i = 1; i <= N; i++) {
a[i] = sc.nextInt();
}
//l存放上一个比它大的元素,r存放下一个比它大的元素
int[] l = new int[N+1];
int[] r = new int[N+1];
//定义栈
Stack<Integer> stack = new Stack<Integer>();
//找上一个比它大的元素
for (int i = 1; i <= N; i++) {
while (!stack.isEmpty() && a[i] > a[stack.peek()]){
stack.pop();
}
if(stack.isEmpty()){
l[i] = -1;
}else {
l[i] = stack.peek();
}
stack.push(i);
}
//栈清空
stack.clear();
//找下一个比它大的元素
for (int i = N; i >= 1; i--) {
while (!stack.isEmpty() && a[i] > a[stack.peek()]){
stack.pop();
}
if(stack.isEmpty()){
r[i] = -1;
}else {
r[i] = stack.peek();
}
stack.push(i);
}
for (int i = 1; i <= N; i++) {
System.out.print(l[i] + " ");
}
System.out.println();
for (int i = 1; i <= N; i++) {
System.out.print(r[i] + " ");
}
}
}
这道题Scanner只能过80%的数据,需要使用快读提高效率,快读可以达到100%:
import java.io.*;
import java.util.Stack;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
public static void main(String[] args)throws IOException {
int N = Integer.parseInt(br.readLine());
//a存放原数组
int[] a = new int[N+1];
String[] s = br.readLine().split(" ");
//l存放上一个比它大的元素,r存放下一个比它大的元素
int[] l = new int[N+1];
int[] r = new int[N+1];
//定义栈
Stack<Integer> stack = new Stack<Integer>();
//找上一个比它大的元素
for (int i = 1; i <= N; i++) {
a[i] = Integer.parseInt(s[i-1]);
while (!stack.isEmpty() && a[i] > a[stack.peek()]){
stack.pop();
}
if(stack.isEmpty()){
l[i] = -1;
}else {
l[i] = stack.peek();
}
stack.push(i);
}
//栈清空
stack.clear();
//找下一个比它大的元素
for (int i = N; i >= 1; i--) {
while (!stack.isEmpty() && a[i] > a[stack.peek()]){
stack.pop();
}
if(stack.isEmpty()){
r[i] = -1;
}else {
r[i] = stack.peek();
}
stack.push(i);
}
for (int i = 1; i <= N; i++) {
out.print(l[i] + " ");
}
out.println();
for (int i = 1; i <= N; i++) {
out.print(r[i] + " ");
}
out.flush();
}
}
学习栈的时候,首先了解栈的定义,栈可以实现什么功能,再用数组进行步骤功能的实现,如果在做题时可以抽象成“先进后出”功能的可以先考虑栈,如果是找出左边/右边第一个比自身大/小的元素,首先考虑单调栈。