思路:看到子数组内的最小值->
dp得到以arr[i]结尾/开始所有子数组的最小值的和
即:leetcode907. 子数组的最小值之和(中等)
https://blog.csdn.net/zhangjiaji111/article/details/124974780
思路转换:
arr[i]贡献:arr[i] * 包括arr[i]的子数组中最小值的和
转化为:arr[i] * (所有子数组的和 - 不包括arr[i]的子数组的最小值的和)
设l[i]表示以arr[i]所有结尾子数组的最小值的和
r[i]表示以arr[i]开始所有子数组最小值的和
则转化为:arr[i]*(sum(l[i]/r[i])-sum(l[0]…l[i-1])-sum(r[i+1]…r[n-1])) 即l和r数组分别用前缀和和后缀和。
具体细节:规定->相等的元素 下标小的较小
ldp[i]=ldp[j:左边第一个小于等于的下标]+(i - j)*arr[i]
rdp[i]=rdp[k:右边第一个小于的下标]+(k - i)*arr[i]
右边第一个小于的在while里面才能得到,所以应该先把r数组得到,再得到rdp数组
class Solution {
public:
int totalStrength(vector<int>& arr) {
const int mod = 1e9 + 7;
int n = arr.size();
vector<long long> ldp(n), rdp(n), r(n, n), stk(n); //
int top = -1;
for (int i = 0; i < n; ++i) {
while (top != -1 && arr[stk[top]] > arr[i]) {
r[stk[top]] = i;
--top;
}
ldp[i] = (top == -1) ? 1ll * (i + 1) * arr[i] : ldp[stk[top]] + 1ll * (i - stk[top]) * arr[i];
ldp[i] %= mod;
stk[++top] = i;
}
for (int i = n - 1; i >= 0; --i) {
rdp[i] = ((r[i] == n) ? 1ll * (n - i) * arr[i] : rdp[r[i]] + 1ll * (r[i] - i) * arr[i]);
rdp[i] %= mod;
}
long long ans = 0, sum = 0;
for (auto& a : ldp) {
sum = (sum + a % mod) % mod;
}
for (int i = 1; i < n; ++i) {
ldp[i] = (ldp[i - 1] + ldp[i] % mod) % mod;
}
for (int i = n - 2; i >= 0; --i) {
rdp[i] = (rdp[i + 1] + rdp[i] % mod) % mod;
}
for (int i = 0; i < n; ++i) {
ans = ans + 1ll * arr[i] * (sum - (i-1>=0?ldp[i-1]:0) - (i+1<n?rdp[i+1]:0) + mod) % mod;
//加个 + mod
ans %= mod;
}
return ans;
}
};
易错点: 1:ldp和rdp数组元素都为long long,因为这道题元素是10的5次方,数组长度为10的9次方。 2:单调栈数组r初始值为n 3:(a-b)%mod -> 要加 +mod