码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode907 Sum of Subarray Minimums

时间:2019-09-09 19:24:24      阅读:91      评论:0      收藏:0      [点我收藏+]

标签:pre   nim   数字   实现   class   复杂度   复杂   tor   ast   

思路:

对于每个数字A[i],使用单调栈找到A[i]作为最小值的所有区间数量,相乘并累加结果。时间复杂度O(n)。

实现:

 1 class Solution
 2 {
 3 public:
 4     int sumSubarrayMins(vector<int>& A)
 5     {
 6         int res = 0;
 7         stack<int> st;
 8         int n = A.size();
 9         const int MOD = 1e9 + 7;
10         for (int i = 0; i < n; i++)
11         {
12             while (!st.empty() && A[i] < A[st.top()])
13             {
14                 int tmp = st.top(); st.pop();
15                 int last = st.empty() ? -1 : st.top();
16                 res = (res + (i - tmp) * (tmp - last) % MOD * A[tmp] % MOD) % MOD;
17             }
18             st.push(i);
19         }
20         while (!st.empty())
21         {
22             int tmp = st.top(); st.pop();
23             int last = st.empty() ? -1 : st.top();
24             res = (res + (n - tmp) * (tmp - last) % MOD * A[tmp] % MOD) % MOD;
25         }
26         return res;
27     }
28 }

leetcode907 Sum of Subarray Minimums

标签:pre   nim   数字   实现   class   复杂度   复杂   tor   ast   

原文地址:https://www.cnblogs.com/wangyiming/p/11493402.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!