栈与队列 - 数组模拟 （Java）

栈与队列 - 数组

数组模拟 - 栈

public class Main {
    static final int N = 100010;
    static int[] stk = new int[N];
    // tt表示栈顶，开始为0
    static int tt = 0;

    // 向栈顶插入一个数
    static void push(int x) {
        tt++;stk[tt] = x;
    }

    // 从栈顶弹出一个数
    static void pop() {
        tt--;
    }

    // 获得栈顶元素
    static int top() {
        return stk[tt];
    }

    // 判断栈是否为空
    static boolean isEmpty() {
        // tt > 0 不空
        return tt > 0;
    }

    public static void main(String[] args) {
        // 测试示例
        push(5);push(10);push(15);
        System.out.println(top()); // 输出：15
        pop(); // 从栈顶弹出15
        System.out.println(top()); // 输出：10
        System.out.println(isEmpty()); // 输出：false
    }
}

数组模拟 - 队列

public class Main {
    static final int N = 100010;
    static int[] q = new int[N];
    // hh 表示队头，tt表示队尾
    static int hh = 0, tt = -1;

    // 向队尾插入一个数
    static void push(int x) {
        tt++;q[tt] = x;
    }

    // 从队头弹出一个数
    static void pop() {
        hh++;
    }

    // 获得队头元素
    static int front() {
        return q[hh];
    }

    // 判断队列是否为空
    static boolean isEmpty() {
        return hh < tt;
    }

    public static void main(String[] args) {
        // 测试示例
        push(5);push(10);push(15);
        System.out.println(front()); // 输出：5
        pop(); // 从对头弹出5
        System.out.println(front()); // 输出：10
        System.out.println(isEmpty()); // 输出：false
    }
}

单调栈

常见模型：找出每个数左边离它最近的比它大/小的

示例代码：Acwing 830 单调栈

public class Main {
    static final int N = 10010;
    static int [] stk= new int [N];
    // tt 表示栈顶，开始为0
    static int tt = 0;
    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();
            // 栈的元素单调递增
            while (tt > 0 && stk[tt] >= x) tt--;
            
            if (tt > 0) System.out.print(stk[tt] + " ");
            else System.out.print("-1 ");
            
            // 将元素存入栈中
            tt++;stk[tt] = x;
        }
    }
}

// 样例 
5
3 4 2 7 5
// 答案
-1 3 -1 2 2 

单调队列 - 滑动窗口

常见模型：找出滑动窗口中的最大值/最小值

示例代码：Acwing 154 滑动窗口

import java.util.Scanner;

public class Main {
        static final int N = 1000010;
        static int [] a = new int[N];
        static int [] q = new int[N];
        static int n,k;
        
    public static void main(String[] args) { 
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        k = sc.nextInt();
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        
        // 滑动窗口最小值
        // hh 表示对头，tt 表示队尾
        int hh = 0;int tt = -1;
        for(int i = 0;i < n;i++) {
            // 窗口队列长度>=k,移动一格
            if(hh<=tt&&i-k+1>q[hh]) hh++;
            // 保证窗口队列单调递减
            while(hh<=tt&&a[q[tt]]>=a[i]) tt--;
            // 将元素存入队列
            tt++;q[tt] = i;
            if (i+1 >= k) {
              System.out.print(a[q[hh]]+" ");
            }
        }
        System.out.println();
        
        // 滑动窗口最大值
        hh = 0;tt = -1;
        for(int i = 0;i < n;i++) {
            if(hh<=tt&&i-k+1>q[hh]) hh++;
            // 保证窗口队列单调递增
            while(hh<=tt&&a[q[tt]]<=a[i]) tt--;
            tt++;q[tt] = i;
            if (i+1 >= k) {
              System.out.print(a[q[hh]]+" ");
            }
        }
    }
}


// 样例
8
3 1 3 -1 -3 5 3 6 7
// 答案
-1 -3 -3 -3 3 3 
3 3 5 5 6 7