基本思想:
和之前的题目一样,但是是升级版;
这个问题能转换一下:
对于所有可能序列的最大最小差,其实是所有序列的最大值和减去所有序列的最小值和;
因此可以转换为两次单调栈问题;
基本代码:
using ll=long long;
class Solution {
public:
ll subArrayRanges(vector<int>& nums) {
ll ret=0;
int n=nums.size();
vector<int>minx(n,0);
vector<int>maxn(n,0);
nums.push_back(INT_MIN);
stack<int>minst;
stack<int>maxst;
for(int i=0;i<nums.size();i++){
while(!minst.empty()&&nums[minst.top()]>=nums[i]){
int index=minst.top();
minst.pop();
int right=i-index;
int l=minst.empty()?-1:minst.top();
int left=index-l;
minx[index]=left*right;
}
minst.push(i);
}
nums.pop_back();
nums.push_back(INT_MAX);
for(int i=0;i<nums.size();i++){
while(!maxst.empty()&&nums[maxst.top()]<=nums[i]){
int index=maxst.top();
maxst.pop();
int right=i-index;
int l=maxst.empty()?-1:maxst.top();
int left=index-l;
maxn[index]=left*right;
}
maxst.push(i);
}
for(int i=0;i<minx.size();i++){
ret+=ll(nums[i])*maxn[i];
ret-=ll(nums[i])*minx[i];
}
return ret;
}
};