AcWing 第九周周赛 题解
3780. 构造数组
3781. 给定一个长度为 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≤109。
输入样例1:
5
1 2 3 2 1
输出样例1:
1 2 3 2 1
输入样例2:
3
10 6 8
输出样例2:
10 6 6
根据题目意思我们可以推出它构造出来的数组最多就只有一个凸峰或者就是直接单调。如果a数组最终有一个凹型曲线,这该a数组不符合题目意思。
代码详解如下:
#include<bits/stdc++.h>
using namespace std;
const int N=500100;
typedef long long ll;
int n,w[N];
ll l[N],r[N];
int stk[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 res=0,k=0;
for(int i=1;i<=n;i++){
ll t=l[i] + r[i+1];//以该点作为最高峰点的总值;
if(t > res) res = t, k = i;
}
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++){
printf("%d ",w[i]);
}
return 0;
}