算法-栈(数组实现)

栈的模拟

首先,我们需要了解什么是栈,栈究竟实现了什么样的模型,有个大致的了解,就可以开始用数组进行栈的功能的实现了(当然,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();
    }
}

学习栈的时候,首先了解栈的定义,栈可以实现什么功能,再用数组进行步骤功能的实现,如果在做题时可以抽象成“先进后出”功能的可以先考虑栈,如果是找出左边/右边第一个比自身大/小的元素,首先考虑单调栈。

  • 7
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值