算法-单调栈-1

题目: 找出一组数中 右边第一个比他大的数 和 左边第一个比他大的数

维护一个栈底到栈顶 从大到小的栈,当压入的数大于栈顶的数 统计栈顶数的左边最大与右边最大

当没有压入值的时候 依次弹出 求出左边最大的值

 

栈里 底下的数是大于当前数的 上面的数是小于当前数的

 

求最近的大于当前值的数是 栈顶到栈底 由大到小

求最近的小雨当前值的数是 栈顶到栈底 由小到大

 

package com.example.demo.algorithm.forceAttempt;

import java.util.Stack;

public class NearMinNum {
    public int[][] getNearMinNum(int[] arr) {
        Stack<Integer> stack = new Stack<>();
        int[][] res = new int[arr.length][2];
        for (int i = 0; i < arr.length; i++) {
            while (!stack.isEmpty() && arr[stack.peek()] > arr[i]) {
                Integer pop = stack.pop();
                int p = stack.isEmpty() ? -1 : stack.peek();
                res[pop][0] = p;
                res[pop][1] = i;
            }
            stack.push(i);
        }
        while (!stack.isEmpty()) {
            Integer pop = stack.pop();
            res[pop][1] = -1;
            int p = stack.isEmpty() ? -1 : stack.peek();
            res[pop][0] = p;
        }
        return res;
    }
}