leetcode做题笔记907. 子数组的最小值之和

给定一个整数数组 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

思路一:模拟题意

c++解法

class Solution {
public:
    int sumSubarrayMins(vector& arr) {
        int n = arr.size(), mod = 1e9 + 7, ans = 0;
        vector s = {-1};
        for(int i = 0; i < n; ++i){
            while(s.size() > 1 && arr[s.back()] >= arr[i]){
                int cur = s.back(); s.pop_back();
                ans = (ans + 1LL * (i - cur) * (cur - s.back()) * arr[cur]) % mod; 
            }
            s.push_back(i);
        }
        while(s.size() > 1){
            int cur = s.back(); s.pop_back();
            ans = (ans + 1LL * (n - cur) * (cur - s.back()) * arr[cur]) % mod;
        }
        return ans;
    }
};

分析:

本题与前日的每日一题相似,需不断比较得到当前子数组中最小值并计算总和,利用数组存放递增的数,最后根据公式 ans = (ans + 1LL * (i - cur) * (cur - s.back()) * arr[cur]) % mod; 计算得到答案

总结:

本题考察对数组的应用,理解题意中前数对子数组的作用即可解决问题

你可能感兴趣的:(数组的应用,leetcode,笔记,算法)