给定一个整数数组 A,找到 min(B) 的总和,其中 B 的范围为 A 的每个(连续)子数组。
由于答案可能很大,因此返回答案模 10^9 + 7。
示例:
输入:[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。
提示:
1 <= A <= 30000
1 <= A[i] <= 30000
思路:
首先要知道STL迭代器的一些知识,
1.map<int, char>::iterator it,it->first 指的就是it所指的迭代器的第一个元素,即int类型的元素(map<int, char>每个迭代器两个元素,一个是int类型,一个是char类型),同理,it->second就是第二个元素,char类型的元素。
2.upper_bound返回的是最后一个大于等于val的位置,也是有一个新元素val进来时的插入位置。
然后膜一下大神的代码
class Solution {
public:
int sumSubarrayMins(vector<int>& A)
{
int size = A.size();
map<int, long long> dp;
long long res = 0,temp=0;
for (int i = 0; i < size; ++i)
{
if (dp.find(A[i]) == dp.end())
dp[A[i]] = 1;
else
dp[A[i]]++;
temp += A[i];
for (map<int, long long>::iterator it = dp.upper_bound(A[i]); it != dp.end();)
{
temp -= (it->first - A[i])*it->second;
dp[A[i]] += it->second;
dp.erase(it++);
}
res = (temp + res) % 1000000007;
}
return res;
}
};