给你一个整数数组 nums 。nums 中,子数组的 范围 是子数组中最大元素和最小元素的差值。
返回 nums 中 所有 子数组范围的 和 。
子数组是数组中一个连续 非空 的元素序列。
示例 1:
输入:nums = [1,2,3]
输出:4
解释:nums 的 6 个子数组如下所示:
[1],范围 = 最大 - 最小 = 1 - 1 = 0
[2],范围 = 2 - 2 = 0
[3],范围 = 3 - 3 = 0
[1,2],范围 = 2 - 1 = 1
[2,3],范围 = 3 - 2 = 1
[1,2,3],范围 = 3 - 1 = 2
所有范围的和是 0 + 0 + 0 + 1 + 1 + 2 = 4
示例 2:
输入:nums = [1,3,3]
输出:4
解释:nums 的 6 个子数组如下所示:
[1],范围 = 最大 - 最小 = 1 - 1 = 0
[3],范围 = 3 - 3 = 0
[3],范围 = 3 - 3 = 0
[1,3],范围 = 3 - 1 = 2
[3,3],范围 = 3 - 3 = 0
[1,3,3],范围 = 3 - 1 = 2
所有范围的和是 0 + 0 + 0 + 2 + 0 + 2 = 4
示例 3:
输入:nums = [4,-2,-3,4,1]
输出:59
解释:nums 中所有子数组范围的和是 59
提示:
1 <= nums.length <= 1000
-109 <= nums[i] <= 109
进阶:你可以设计一种时间复杂度为 O(n) 的解决方案吗?
解题思路;
通过单调栈解决问题,任意一个点可能是数组的最小值或者最大值,通过单调栈计算影响范围。假设 nums[i] 左侧最近的比它小的数为 nums[j],右侧最近的比它小的数为nums[k],那么所有以nums[i] 为最小值的子数组数目为 nums[i] * (k−i) *(i−j)
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/sum-of-subarray-ranges/solution/zi-shu-zu-fan-wei-he-by-leetcode-solutio-lamr/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码如下
class Solution {
public:
long long subArrayRanges(vector<int>& nums) {
int len = nums.size();
//minleft,minright确定当前值作为最小值的范围,maxleft,maxright确定作为当前最大值的范围
vector<int>minleft(len),minright(len),maxleft(len),maxright(len);
//stack<int>addstack 维护递增栈,substack维护递减栈
stack<int>addstack,substack;
//1 从左向右遍历数据 (相等 序号小的更小)
//1.1 通过addstack求maxleft,如果当前元素比栈顶元素小,弹出栈顶元素,最后栈顶元素就是当前元素左边最大值minleft。
//1.2 通过substack求minleft,如果当前元素比栈顶元素大,弹出栈顶元素,最后栈顶元素就是当前元素的左边最小值maxleft。
for(int i = 0; i < len; ++i) {
while (!addstack.empty(<