题目链接:I-Max answer
题目大意:给出n个数,求得一个区间使得该区间最小的数乘上该区间所有数的和最大,即,输出该结果。注意:a[i]可为负。
思路:若a[i]全为正,可参考poj2796,单调栈的经典应用。具体来考虑单调栈的作用,它可以求出以每个a[i]为最小值时所对应的最大区间(l,r),区间和用前缀和计算即可:sum[i]-sum[j],若a[i]为正,显然直接维护答案即可;
若a[i]为负,考虑区间和最小,即找到在以a[i]对应的最大区间范围(l.r)内最小的sum[i]和最大的sum[j],问题再次转化为在区间内求最值,考虑到数据范围即用st表O(nlogn)预处理,O(1)区间查询即可。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cmath>
typedef long long ll;
using namespace std;
#define INF 0x3f3f3f3f
const int mod=1e9+7;
const int maxn=5e5+10;
int n,a[maxn],l[maxn],r[maxn];
ll stmx[maxn][23],stmn[maxn][23];
ll sum[maxn];
stack<int>s;
void st(int n){
for(int i=0;i<=n;i++)stmx[i][0]=stmn[i][0]=sum[i];
for(int j=1;(1<<j)<=n+1;j++)
{
for(int i=0;i+(1<<j)-1<=n;i++)
{
stmx[i][j]=max(stmx[i][j-1],stmx[i+(1<<(j-1))][j-1]);
stmn[i][j]=min(stmn[i][j-1],stmn[i+(1<<(j-1))][j-1]);
}
}
}
ll askma(int l,int r)
{
int k=log2(r-l+1);
return max(stmx[l][k],stmx[r-(1<<k)+1][k]);
}
ll askmi(int l,int r)
{
int k=log2(r-l+1);
return min(stmn[l][k],stmn[r-(1<<k)+1][k]);
}
void cal(){
for(int i=1;i<=n;i++){
if(s.empty())s.push(i);
else{
while(!s.empty()&&a[s.top()]>=a[i])s.pop();
if(s.empty())l[i]=1;else l[i]=s.top()+1;
s.push(i);
}
}
while(s.size())s.pop();
for(int i=n;i>=1;i--){
if(s.empty())s.push(i);
else{
while(!s.empty()&&a[s.top()]>=a[i])s.pop();
if(s.empty())r[i]=n;else r[i]=s.top()-1;
s.push(i);
}
}
}
int main()
{
scanf("%d",&n);
sum[0]=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
sum[i]=sum[i-1]+1ll*a[i];
l[i]=r[i]=i;
}
st(n);
cal();
ll tmp,ans=-1e18;
for(int i=1;i<=n;i++){
if(a[i]<0){
ll x=askmi(i,r[i]);
ll y=askma(l[i]-1,i-1);
tmp=max(ans,(x-y)*(ll)a[i]);
}else
if(a[i]>0){
ll x=askma(i,r[i]);
ll y=askmi(l[i]-1,i-1);
tmp=max(ans,(x-y)*(ll)a[i]);
}else tmp=max(ans,0ll);
ans=max(tmp,ans);
}
printf("%lld\n",ans );
return 0;
}
通过这道题顺便学习单调栈的具体写法(两个for),并再次加深st表的理解。