题意:
给你一个整数数组 nums (下标从 0 开始)和一个整数 k 。
一个子数组 (i, j) 的 分数 定义为 min(nums[i], nums[i+1], ..., nums[j]) * (j - i + 1) 。
一个 好 子数组的两个端点下标需要满足 i <= k <= j 。
请你返回 好 子数组的最大可能 分数 。
数据范围:
1 <= nums.length <= 1e5
1 <= nums[i] <= 2 * 1e4
0 <= k < nums.length
解法:
单调栈计算出每个a[i]作为最小值所能扩展的最左下标l[i],和最优下标r[i].
如果l[i]<=k<=r[i],那么可以用(r[i]-l[i]+1)*a[i]更新答案.
code:
const int maxm=1e5+5;
int stk[maxm];
int l[maxm],r[maxm];
class Solution {
public:
int maximumScore(vector<int>& a, int k) {
int n=a.size();
a.insert(a.begin(),-1);
a.push_back(-1);
k++;
int head=0;
for(int i=1;i<=n+1;i++){
while(head&&a[stk[head]]>a[i]){
r[stk[head]]=i-1;
head--;
}
stk[++head]=i;
}
head=0;
for(int i=n;i>=0;i--){
while(head&&a[stk[head]]>a[i]){
l[stk[head]]=i+1;
head--;
}
stk[++head]=i;
}
int ans=0;
for(int i=1;i<=n;i++){
if(l[i]>k||r[i]<k)continue;
ans=max(ans,(r[i]-l[i]+1)*a[i]);
}
return ans;
}
};