C-sequence【线段树+单调栈】

找出左边第一个比ai小的数的位置li与右边第一个比ai小的数的位置ri

#include<bits/stdc++.h>
using namespace std;
#define maxn 3000010
typedef long long ll;
ll INF=1e18;
struct Tree{
	int l,r;
	ll mi,mx;
}tree[maxn<<2];
int a[maxn],b[maxn],l[maxn],r[maxn];ll sum[maxn];
stack<int>sta;
void pushup(int k){
	tree[k].mi=min(tree[k<<1].mi,tree[k<<1|1].mi);
	tree[k].mx=max(tree[k<<1].mx,tree[k<<1|1].mx);
}
void build(int l,int r,int k){
	tree[k].l=l; tree[k].r=r; 
	if(l==r){
		tree[k].mi=sum[l];
		tree[k].mx=sum[l];
		return;
	}
	int mid=(l+r)>>1;
	build(l,mid,k<<1);
	build(mid+1,r,k<<1|1);
	pushup(k);
}
ll querymi(int l,int r,int k){
	if(l > r) return INF;
	if(tree[k].l>r || tree[k].r<l) return INF;
	if(l<=tree[k].l&&tree[k].r<=r) return tree[k].mi;
	int mid=(tree[k].l+tree[k].r)>>1; ll res=INF;
	if(l<=mid) res=min(res,querymi(l,r,k<<1));
	if(r> mid) res=min(res,querymi(l,r,k<<1|1));
	return res;
}
ll querymx(int l,int r,int k){
	if(l > r) return -INF;
	if(tree[k].l>r || tree[k].r<l) return -INF;
	if(l<=tree[k].l&&tree[k].r<=r) return tree[k].mx;
	int mid=(tree[k].l+tree[k].r)>>1; ll res=-INF;
	if(l<=mid) res=max(res,querymx(l,r,k<<1));
	if(r> mid) res=max(res,querymx(l,r,k<<1|1));
	return res;
}
int main(){
	int n; scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),l[i]=0,r[i]=0;
	for(int i=1;i<=n;i++){
		scanf("%d",&b[i]);
		sum[i]=sum[i-1]+b[i];
	}
	build(1,n,1);
	for(int i=1;i<=n;i++){
		while(!sta.empty() && a[sta.top()]>=a[i]) sta.pop();
		if(sta.empty()) l[i]=0;
		else l[i]=sta.top();
		sta.push(i);
	}
	while(!sta.empty()) sta.pop();
	for(int i=n;i>=1;i--){
		while(!sta.empty() && a[sta.top()]>=a[i]) sta.pop();
		if(sta.empty()) r[i]=n+1;
		else r[i]=sta.top();
		sta.push(i);
	}
	for(int i=1;i<=n;i++) l[i]+=1,r[i]-=1;
	ll ans=-INF;
	for(int i=1;i<=n;i++){
		if(a[i]>0) ans=max(ans,1ll*a[i]*(querymx(i,r[i],1)-querymi(l[i]-1,i-1,1)));
		else ans=max(ans,1ll*a[i]*(querymi(i,r[i],1)-querymx(l[i]-1,i-1,1)));
	}
	printf("%lld\n",ans);
	return 0;
}