CF1313C2-单调栈

CF1313C2

题目描述

题解

单调栈
显然整个大楼应呈现“山”字形，于是我们考虑求出每一个位置左右两边为单调序列的情况（单调栈实现），最后统计答案

代码

#include<bits/stdc++.h>
#define int long long
#define M 500009
using namespace std;
int read(){
	int f=1,re=0;char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
int n,a[M],q[M],head=1,tail,maxn,pos,l[M],r[M],pre;
signed main(){
	n=read();
	for(int i=1;i<=n;i++)a[i]=read();
	tail=0,head=1;
	for(int i=1;i<=n;i++){
		while(a[q[tail]]>a[i]&&head<=tail) tail--;
		int now=q[tail];
		l[i]=l[now]+(i-now)*a[i];
		q[++tail]=i;
	}head=1,tail=0;
	q[++tail]=n+1;//关键
	for(int i=n;i>=1;i--){
		while(a[q[tail]]>a[i]&&head<=tail) tail--;
		int now=q[tail];
		r[i]=r[now]+(now-i)*a[i];
		q[++tail]=i;
	}
	//for(int i=1;i<=n;i++) printf("%lld %lld\n",l[i],r[i]);
	for(int i=1;i<=n;i++)
		if(l[i]+r[i]-a[i]>maxn){
			maxn=l[i]+r[i]-a[i];
			pos=i;
		}//printf("%lld\n",pos);
		pre=a[pos];
	for(int i=pos;i>=1;i--){//修改左边的非法大楼高度
		if(a[i]>pre) a[i]=pre;
		pre=a[i];
	}pre=a[pos];
	for(int i=pos;i<=n;i++){//同上右边
		if(a[i]>pre) a[i]=pre;
		pre=a[i];
	}for(int i=1;i<=n;i++) printf("%lld ",a[i]);
	printf("\n");
	return 0;
}