单调栈结构【Java实现】

题目:给定一个不含有重复值的数组arr,找到每一个 i 位置左边和右边离 i 位置最近且值比 arr[i] 小的位置。返回所有位置相应的信息。

举例: arr = [3,4,1,5,6,2,7] 返回如下二维数组作为结果: { {-1,2}, {0,2}, {-1,-1}, {2,5}, {3,5}, {2,-1}, {5,-1} }

思路一:(复杂度O(N2))

先整体for循环遍历数组中每一个数,然后再找小于它的数的下标。具体说明见代码注释:

//每一个位置分别向左或向右遍历以下
    public static int[][] rightWay(int[] arr) {
        int[][] res = new int[arr.length][2]; //初始化返回数组
        //遍历数组中每一个数,向左向右寻找小于它的下标
        for (int i = 0;i < arr.length; i++) {
            int left = -1;
            int right = -1;
            int cur = i - 1;//获取当前位置左边的下标
            //先向左遍历 下标大于0时循环
            while (cur >= 0) {
                if (arr[cur] < arr[i]) {  //如果cur对应的值小于arr[i],就把cur赋值给left,找到就break跳出循环
                    left = cur;
                    break;
                }
                cur--;//没找到就继续向左找
            }
            cur = i + 1; //开始找右边的 下标小于数组长度时循环 其他同上
            while(cur < arr.length) {
                if (arr[cur] < arr[i]) {
                    right = cur;
                    break;
                }
                cur++;
            }
            res[i][0] = left;
            res[i][1] = right;
        }
        return res;
    }

测试结果:

思路二:使用单调栈结构

准备一个栈,战中存放的是数组的位置。

求解流程:

代码实现:

public static int[][] getNearLessNoRepeat(int[] arr) {
        int[][] res = new int[arr.length][2];
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < arr.length; i++) {
            //如果当前遍历到的值比栈顶元素代表的值小,就破坏了单调性,需要弹出
            while (!stack.isEmpty() && arr[stack.peek()] > arr[i]) {
                int popIndex = stack.pop();
                int left = stack.isEmpty() ? -1 : stack.peek();//如果为空,就是-1,否则就是栈顶元素
                res[popIndex][0] = left;
                res[popIndex][1] = i;
            }
            //不需要弹出的时候 i入栈
            stack.push(i);
        }
        //进入最后清算阶段
        while (!stack.isEmpty()) {
            int popIndex = stack.pop();
            int left = stack.isEmpty() ? -1 : stack.peek();
            res[popIndex][0] = left;
            res[popIndex][1] = -1;
        }
        return res;
    }

测试结果:

经验分享 程序员 微信小程序 职场和发展