LeetCode 1793. 好子数组的最大分数(单调栈)

题意:
给你一个整数数组 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;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值