前言
前几天群里吹水,有同学去面头条,在群里分享面试题,其实就有一道单调栈,碰巧前几天我也刷了同样一道,但是当时没想到写O(N)的解法,吹完水就去了解了一下这个所谓的单调栈。
单调栈
针对求下个更大的数之类题目 的一个很巧妙地解法,时间复杂度可以达到O(N)
Input: [73, 74, 75, 71, 69, 72, 76, 73]
Output: [1, 1, 4, 2, 1, 1, 0, 0]
题目很简单,就是求指定数右边的第一个比该数大的数字的下标和该数下标的距离,不存在下个更大的数的就拿0代替
public int[] dailyTemperatures(int[] T) {
int n = T.length;
int[] dist = new int[n];
Stack<Integer> indexs = new Stack<>();
for (int curIndex = n-1; curIndex >= 0; curIndex--) {
// 如果当前遍历到的数字比栈顶元素大,就pop,直到找到比当前元素大的数字
while (!indexs.isEmpty() && T[curIndex] >= T[indexs.peek()]) {
indexs.pop();
}
if(!indexs.isEmpty()) dist[curIndex] =indexs.peek()- curIndex;
indexs.push(curIndex);
}
return dist;
}
用栈来保存离当前遍历到的元素最近且比该元素大的数字下标。
栈为空的话直接入栈
若当前数字小于栈顶元素对应数字,直接入栈; 否则的话,栈进行pop直到找到比当前数字大的元素对应数(栈内保存下标)
公众号