题目:给定一个不含有重复值的数组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;
}
测试结果: