leetcode2281.巫师的总力量和(周赛,困难)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
思路:看到子数组内的最小值->
dp得到以arr[i]结尾/开始所有子数组的最小值的和

即:leetcode907. 子数组的最小值之和(中等)
https://blog.csdn.net/zhangjiaji111/article/details/124974780
思路转换
arr[i]贡献:arr[i] * 包括arr[i]的子数组中最小值的和
转化为:arr[i] * (所有子数组的和 - 不包括arr[i]的子数组的最小值的和)
设l[i]表示以arr[i]所有结尾子数组的最小值的和
r[i]表示以arr[i]开始所有子数组最小值的和
则转化为:arr[i]*(sum(l[i]/r[i])-sum(l[0]…l[i-1])-sum(r[i+1]…r[n-1])) 即l和r数组分别用前缀和和后缀和

具体细节:规定->相等的元素 下标小的较小
ldp[i]=ldp[j:左边第一个小于等于的下标]+(i - j)*arr[i]
rdp[i]=rdp[k:右边第一个小于的下标]+(k - i)*arr[i]
右边第一个小于的在while里面才能得到,所以应该先把r数组得到,再得到rdp数组

class Solution {
public:
    
    int totalStrength(vector<int>& arr) {
 
        const int mod = 1e9 + 7;
        int n = arr.size();
        vector<long long> ldp(n), rdp(n), r(n, n), stk(n); //
        int top = -1;
        for (int i = 0; i < n; ++i) {
            while (top != -1 && arr[stk[top]] > arr[i]) {
                r[stk[top]] = i; 
                --top;
            }
            ldp[i] = (top == -1) ? 1ll * (i + 1) * arr[i] : ldp[stk[top]] + 1ll * (i - stk[top]) * arr[i];
            ldp[i] %= mod;
            stk[++top] = i;
        }
        for (int i = n - 1; i >= 0; --i) {
            rdp[i] = ((r[i] == n) ? 1ll * (n - i) * arr[i] : rdp[r[i]] + 1ll * (r[i] - i) * arr[i]);
            rdp[i] %= mod;
        }
        long long ans = 0, sum = 0;
        for (auto& a : ldp) {
            sum = (sum + a % mod) % mod; 
        }
        for (int i = 1; i < n; ++i) {
            ldp[i] = (ldp[i - 1] + ldp[i] % mod) % mod;
        }
        for (int i = n - 2; i >= 0; --i) {
            rdp[i] = (rdp[i + 1] + rdp[i] % mod) % mod;
        }
        for (int i = 0; i < n; ++i) {
            ans = ans + 1ll * arr[i] * (sum - (i-1>=0?ldp[i-1]:0) - (i+1<n?rdp[i+1]:0) + mod) % mod;
            //加个 + mod
            ans %= mod;
        }
        return ans;
    }
};
易错点: 1:ldp和rdp数组元素都为long long,因为这道题元素是10的5次方,数组长度为10的9次方。 2:单调栈数组r初始值为n 3:(a-b)%mod -> 要加 +mod

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值