给定一个长度为 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×105,1≤mi≤10^9。
输入样例1:
5
1 2 3 2 1
输出样例1:
1 2 3 2 1
输入样例2:
3
10 6 8
输出样例2:
10 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;
}