[单调栈+前缀和的前缀和] 巫师的总力量和

题目描述

作为国王的统治者，你有一支巫师军队听你指挥。

给你一个下标从 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;
    }
};