单调栈+线段树

链接：https://ac.nowcoder.com/acm/contest/884/C
来源：牛客网
 

sequence

时间限制：C/C++ 3秒，其他语言6秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld

题目描述

Your are given two sequences a1…na_{1 \dots n}a1…n​ and b1…nb_{1 \dots n}b1…n​ .You need to answer max⁡1≤l≤r≤n{min(al…r)×sum(bl…r)}\displaystyle \max_{1 \le l \le r \le n} \{min(a_{l \dots r}) \times sum(b_{l \dots r})\}1≤l≤r≤nmax​{min(al…r​)×sum(bl…r​)} 。

Where min(a) means the minimal value of every element of sequence a, sum(a) means the sum of every element of sequence a .

输入描述:

The first line contains an integer n .

The second line contains n integers meaning a1…na_{1 \dots n}a1…n​ .

The third line contains n integers meaning b1…nb_{1 \dots n}b1…n​ .

输出描述:

An integer meaning the answer.

示例1

输入

复制

3
1 -1 1
1 2 3

输出

复制

3

备注:

For all test datas, 1≤n≤3×106,−106≤ai,bi≤1061 \leq n \leq 3 \times 10^6,-10^6 \leq a_i,b_i \leq 10^61≤n≤3×106,−106≤ai​,bi​≤106.

#include<bits/stdc++.h>
using namespace std;
int a[3000006];
int b[3000006];
int L[3000006],R[3000006];
int n;
stack <int > stk;
//!!!线段树4倍空间 
long long  sum[3000006],mx[4*3000006],mn[4*3000006],Mx,Mn,ans=-1e18;
void up(int o,int l,int r)
{
	if(l==r)
	{
		mx[o]=mn[o]=sum[l];
		return ;
	}
	int mid= (l+r)/2;
	up(o*2,l,mid);
	up(o*2+1,mid+1,r);
	mx[o]=max(mx[o*2],mx[o*2+1]);
	mn[o]=min(mn[o*2],mn[o*2+1]);	
	//同时建两颗线段树 
}
void que(int o,int l,int r,int ql,int qr)//查询区域内最值 
{
	if(ql<=l&&r<=qr)
	{
		Mx=max(Mx,mx[o]);
		Mn=min(Mn,mn[o]);
		return ; 
	}
	int mid= (l+r)/2;
	if(ql<=mid) que(o*2,l,mid,ql,qr);
	if(qr>mid) que(o*2+1,mid+1,r,ql,qr);	
}
void go()
{
	for(int i=1;i<=n;i++)
	{
		if(a[i]>=0)
		{
			Mx=-1e18;
			que(1,0,n,i,R[i]);
			long long t=Mx;//求右边最大前缀和  并保存 
			Mn=1e18;
			que(1,0,n,L[i]-1,i-1);//左边最小前缀和 
			ans=max(ans,a[i]*(t-Mn));
		}
		else 
		{
			Mn=1e18;
			que(1,0,n,i,R[i]);
			long long t=Mn;
			Mx=-1e18;
			que(1,0,n,L[i]-1,i-1);
			ans=max(ans,a[i]*(t-Mx));
		}
	}
}
int main()
{
	scanf("%d",&n) ;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&b[i]);
		sum[i]=sum[i-1]+b[i];
		//求前缀和  此处需要存sum[0]  因为计算前缀和时 需要前移一位 
	}
	for(int i=1;i<=n;i++) 
	{
		while(!stk.empty()&&a[stk.top()]>=a[i])
		{
			stk.pop();
		}
		if(stk.empty())
		{
			L[i]=1;
			stk.push(i);
		}
		else 
		{
			L[i]=stk.top()+1;
			stk.push(i);
		}		
	}
	//单调栈  L[i] 保存a[i] 的最左范围下标 使得在这个范围内 a[i]为最小值 
	
	while(!stk.empty())
	{
		stk.pop();
	}
	//stack无 .clear()  则需要手动清空 
	for(int i=n;i>=1;i--)
	{
		while(!stk.empty()&&a[stk.top()]>=a[i])
		{
			stk.pop();
		}
		if(stk.empty())
		{
			R[i]=n;
		}
		else R[i]=stk.top()-1;
		
		stk.push(i);
	}
	//单调栈  L[i] 保存a[i] 的最左范围下标 使得在这个范围内 a[i]为最小值
	up(1,0,n);//建立  前缀和 最大最小树 
	go();//枚举每个点作为最小值的情况 ，取最大值 
	printf("%lld\n",ans);	
}