构造数组(单调栈,递推,枚举)

给定一个长度为 n 的整数数组 m1,m2,,mn。

现在,请你构造一个数组 a1,a2,,an。

对于构造的数组,有以下三点要求:

∀i∈[1,n]1≤ai≤mi 成立。
∀i∈[1,n], 不存在数对 j,k 同时满足 j<i<k 且 aj>ai<ak。
数组中所有元素之和尽可能大。
请输出任意合理方案。

输入格式
第一行包含整数 n。

第二行包含 n 个整数 m1,m2,,mn。

输出格式
输出 n 个整数,表示你构造出的数组 a1,a2,,an。

如果答案不唯一, 输出任意合理方案均可。

数据范围
前三个测试点满足 1≤n≤10。
全部测试点满足 1≤n≤5×1051≤mi≤10^9。

输入样例15
1 2 3 2 1
输出样例11 2 3 2 1
输入样例23
10 6 8
输出样例210 6 6

算法:
在这里插入图片描述

题目要求的一定的一个单峰函数,如果存在两个山峰,那中间一定会存在值比两边的值小;
1、枚举先找到峰值的位置,找到以某个元素为峰值,把峰值左边构造成单调递增,右边单调递减,然后求最大的总和;所以左右两边是相互独立了,可以分别求。
2、枚举峰值相当于找中间的两个数Li 和 Ri,不管是哪个高都不影响
在这里插入图片描述
所以峰值就是求val = max{Li,Ri}
3、求Li,让Li变为单调递增的最大和,我们可以先递推的求下理论最大值
在这里插入图片描述
但是求这样求每一个ai,然后求和再加回来的时间是O(n),求n个ai时间就变成O(n^2),需要优化;
假设ai = mi,如果mi-1 > mi,那么也只能取mi,mi-2 > mi ,也是取mi,所以理论最大值是可以取的到;
在这里插入图片描述
求每个数左边第一个比它小的数,可以用单调栈来优化;
右边Ri同理来求,整的时间复杂度O(n)

#include <iostream>

using namespace std;

typedef long long LL;

const int N = 500010;

int n;
int w[N],stk[N];
LL l[N],r[N];

int main(){
    scanf("%d",&n);
    for(int i = 1; i <= n; i++) scanf("%d",&w[i]);
    //求左边的数
    int tt = 0;
    for(int i = 1; i <= n; i++){
        while(tt && w[stk[tt]] >= w[i]) tt--;
        l[i] = l[stk[tt]] + (LL)(i - stk[tt]) * w[i];
        stk[++tt] = i;
    }
    
    //求右边的数
    tt = 0, stk[0] = n + 1;
    for(int i = n; i ; i--){
        while(tt && w[stk[tt]] >= w[i]) tt--;
        r[i] = r[stk[tt]] + (LL)(stk[tt] - i) * w[i];
        stk[++tt] = i;
    }
    
    //求总和,确定峰值的位置
    LL sum = 0, k;
    for(int i = 1; i <= n; i++){
        LL t = l[i] + r[i + 1];
        if(t > sum){
            sum = t;
            k = i;
        }
    }
    
    //从后往前推ai的值,由于k 和 k+1 已经固定
    for(int i = k - 1; i; i--) w[i] = min(w[i],w[i+1]);
    for(int i = k + 2; i <= n; i++) w[i] = min(w[i],w[i-1]);
    for(int i = 1; i <= n; i++) cout<<w[i]<<" ";
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值