Leetcode 每日一题10.28 子数组的最小值之和

最新推荐文章于 2026-06-22 19:30:00 发布

原创最新推荐文章于 2026-06-22 19:30:00 发布 · 319 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode #算法 #动态规划

博客围绕给定整数数组，求其每个连续子数组最小值总和并对结果取模的问题展开。通过示例展示问题求解，提出使用单调栈和动态规划的解题思路，将原问题分解为子问题，用 dp[i] 表示以 i 结尾子数组最小值之和，借助单调栈求上一个更小元素下标。

给定一个整数数组 arr，找到 min(b) 的总和，其中 b 的范围为 arr 的每个（连续）子数组。由于答案可能很大，因此返回答案模 10^9 + 7 。

示例 1：

输入：arr = [3,1,2,4]
输出：17
解释：
子数组为 [3]，[1]，[2]，[4]，[3,1]，[1,2]，[2,4]，[3,1,2]，[1,2,4]，[3,1,2,4]。
最小值为 3，1，2，4，1，1，2，1，1，1，和为 17。
示例 2：

输入：arr = [11,81,94,43,3]
输出：444

提示：

1 <= arr.length <= 3 * 104
1 <= arr[i] <= 3 * 104

解题思路

单调栈 + 动态规划。

每个元素 arr[i] 对前面的所有子数组的贡献取决与它的「上一个更小元素」的位置。

假设 arr[i] 的上一个更小元素为 arr[j]，那么在区间 (j, i] 内的所有以 i 结尾的子数组的最小值都为 arr[i]。区间 (j, i] 内共有 (i - j) 个以 i 结尾的子数组。那么区间 (j, i] 内的所有以 i 结尾的子数组的最小值之和为 arr[i] * (i - j)。

所以在区间 [0, i] 内所有以 i 结尾的子数组的最小值之和为 sum = arr[i] * (i - j) + 区间 [0, j] 内所有以 j 结尾的子数组的最小值之和。

可以看出，这里可以将原问题分解为多个子问题。所以想到用动态规划。

我们令 dp[i] 表示以 i 结尾的子数组的最小值之和。那么 dp[i] = dp[j] + arr[i] * (i - j)。

其中，j 为 arr[i] 上一个更小元素的下标，可以利用单调栈求出。

代码如下

C++

class Solution {
public:
    const int MOD = 1e9 + 7;
    int sumSubarrayMins(vector<int>& arr) {
        int n = arr.size();
        // 单调栈：求上一个更小元素
        stack<int> stk;
        // dp[i] 表示以 i 结尾的子数组的最小值之和
        vector<int> dp(n);
        for (int i = 0; i < n; i++) {
            while (!stk.empty() && arr[stk.top()] >= arr[i]) {
                stk.pop(); 
            }    
            // 上一个更小元素
            int preIdx = stk.empty() ? -1 : stk.top();
            stk.push(i);
            
            if (preIdx == -1) {
                // 如果没有上一个更小元素，则以 i 结尾的 i + 1 个区间的最小值都为 arr[i]
                dp[i] = arr[i] * (i + 1); 
            } else {
                // 如果存在上一个更小元素，则 [preIdx, i] 之间的所有子区间的最小值都为 arr[i]
                dp[i] = dp[preIdx] + arr[i] * (i - preIdx);
            }
        }
        int res = 0;
        for (int cnt : dp) {
            res = (res + cnt) % MOD;
        }
        return res;
    }
};