题目描述
作为国王的统治者,你有一支巫师军队听你指挥。
给你一个下标从 0 开始的整数数组 strength ,其中 strength[i] 表示第 i 位巫师的力量值。对于连续的一组巫师(也就是这些巫师的力量值是 strength 的 子数组),总力量 定义为以下两个值的 乘积 :
巫师中 最弱 的能力值。
组中所有巫师的个人力量值 之和 。
请你返回 所有 巫师组的 总 力量之和。由于答案可能很大,请将答案对 109 + 7 取余 后返回。
子数组 是一个数组里 非空 连续子序列。
代码
class Solution {
public:
int totalStrength(vector<int>& v) {
vector<pair<int,int>>st,sq; //值+下标
vector<int> left(v.size()),right(v.size());
using LL=long long;
int Mod=1e9+7;
//找小于等于当前元素的左边的第一个元素的下标
for(int i=0,j=v.size()-1;i<v.size();i++,j--)
{
while(!st.empty()&&st.back().first>=v[i]) st.pop_back(); //大于等于当前元素的左侧都退栈
left[i]=st.empty()?0:st.back().second+1; //找不到为0,找到还要+1 //(防止同一区间被统计:左边可以小于等于)
st.emplace_back(v[i],i);
while(!sq.empty()&&sq.back().first>v[j]) sq.pop_back(); //大于当前元素的右侧都退栈
right[j]=sq.empty()?v.size()-1:sq.back().second-1; //找不到为n-1,找到要-1 //(防止同一区间被统计:右边不可以小于等于)
sq.emplace_back(v[j],j);
}
//求前缀和
vector<int> preSum(v); //先整个拷贝
for(int i=1;i<preSum.size();i++) preSum[i]=(preSum[i]+preSum[i-1])%Mod; //初始化方式
//求前缀和的前缀和
vector<int> prepreSum(preSum); //仍然先整个拷贝
for(int i=1;i<preSum.size();i++) prepreSum[i]=(prepreSum[i]+prepreSum[i-1])%Mod;
LL Sum=0;
for(int i=0;i<preSum.size();i++){
int l=left[i],r=right[i];
LL _1,_2;
_1=(LL)(i-l+1)*(prepreSum[r]-(i-1<0?0:prepreSum[i-1]))%Mod;
if(i-1<0) _2=0;
else _2=(LL)(r-i+1)*(prepreSum[i-1]-(l-2<0?0:prepreSum[l-2]))%Mod;
Sum+=(LL)((_1-_2+Mod)%Mod)*v[i]%Mod;
Sum%=Mod; //它本身可能还是会溢出的
}
return Sum;
}
};