首页 > 其他 > 详细

leetcode907 Sum of Subarray Minimums

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

思路:

对于每个数字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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!