C2. Skyscrapers (hard version)
题目传送门:
C2. Skyscrapers (hard version)
题目大意:
有n栋楼,每栋楼最高为hi,不允许存在一栋楼两边都有比他更高的楼,要求建成的楼的总高度最大。
思路:
1.用单调栈。
延续C1的想法,我们要找到一个最高点,然后向左向右全部都向低处扩展。这个最高点将序列分成了两个部分,从左到右是非递减序列,从右到左是非递减序列。那么我们利用单调栈的思想,求出每个数向左所影响的区间和向右所影响的区间,分别从左往右和从右往左求一次单调栈即可。然后在求的过程中维护前缀楼高和后缀楼高。详细见代码。
AC Code
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=5e5+10;
LL m[N];
LL l[N],r[N];
stack<LL>st;//单调栈
int main()
{
LL n;
scanf("%lld",&n);
for(int i=1;i<=n;i++) scanf("%lld",&m[i]);
for(int i=1;i<=n;i++)
{
while(!st.empty()&&m[i]<m[st.top()])
st.pop();
if(st.empty()) l[i]=(LL)i*m[i];
else l[i]=(LL)(i-st.top())*m[i]+l[st.top()];
st.push(i);
}
while(!st.empty()) st.pop();
for(int i=n;i>=1;i--)
{
<