icpc南昌邀请赛网络赛 I-Max answer

题目链接：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表的理解。