题目
给定一个整数数组 A A A,坡是元组 ( i , j ) (i, j) (i,j),其中 i < j i < j i<j且 A [ i ] < = A [ j ] A[i] <= A[j] A[i]<=A[j]。这样的坡的宽度为 j − i j - i j−i。
找出 A A A 中的坡的最大宽度,如果不存在,返回 0 。
思路
先遍历一遍数组,讲单调递减的元素放入栈中,然后从数组末尾开始遍历,用 e n d end end表示末尾开始的元素 i n d e x index index。当 e n d end end表示的元素不小于栈顶元素的时候,计算一个坡宽,栈中元素出栈。否则 e n d − 1 end-1 end−1。为什么要用单调递减栈,比如,当前栈顶元素为 a a a,所对应的值为 n u m s [ a ] nums[a] nums[a],可能 a a a后面有 n n n个元素都大于 n u m s [ a ] nums[a] nums[a],所以没有入栈,这意味着,如果这些没入栈的元素右边存在一个不小于这些元素的值 n u m s [ b ] nums[b] nums[b],那么 n u m s [ b ] nums[b] nums[b]一定不小于 n u m s [ a ] nums[a] nums[a],所以这些元素就不用考虑了。栈里的其他元素同理。
代码
class Solution {
public:
int maxWidthRamp(vector<int>& nums) {
int res = 0;
stack<int> stk;
for (int i = 0; i < nums.size(); ++i) {
if (stk.empty() || (nums[i] <= nums[stk.top()])) {
stk.push(i);
}
}
int end = nums.size() - 1;
while (!stk.empty() && end >= stk.top()) {
while (!stk.empty() && nums[end] >= nums[stk.top()] && end >= stk.top()) {
res = max(res, end - stk.top());
stk.pop();
}
--end;
}
return res;
}
};