2019牛客多校第四场 C squence 【单调栈+线段树】

题目链接：https://ac.nowcoder.com/acm/contest/884/C

/*
思路：构造一个单调减栈（栈里元素下标增大值增大），把序列 A 添加进去，如果当前 A[i] 比栈头元素小，
将栈头元素弹出，此时以弹出元素为最小的区间就是此时栈头元素的下标和当前 A[i] 下标即i，然后用线段
树来求出这个区间里 B 的前缀和最大的差值。
具体为：假设以当前A[i]为最小的区间是l,r:
如果当前 A[i] 是正数，求区间 i~r 的最大值， l~i-1的最小值；
如果当前 A[i] 是负数，求区间 i~r 的最小值， l~i-1的最大值
*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<vector>
#include<queue>
#include<bitset>
#define max(a,b) a>b?a:b
#define min(a,b) a>b?b:a
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int N=3e6+10;

ll sta[N];
ll A[N],B[N],sum[N];
ll n;
ll tree_max[N*4],tree_min[N*4];

void build_max(ll a,ll b,ll r)
{
    if(a==b)
    {
        tree_max[r]=tree_min[r]=sum[a];
        return;
    }
    ll mid=(a+b)/2;
    build_max(a,mid,r<<1);
    build_max(mid+1,b,r<<1|1);
    tree_max[r]=max(tree_max[r<<1],tree_max[r<<1|1]);
    tree_min[r]=min(tree_min[r<<1],tree_min[r<<1|1]);
}

//查询区间最大值
ll query_max(ll a,ll b,ll r,ll qa,ll qb)
{
    if(a>=qa&&b<=qb)
        return tree_max[r];
    ll ans=-inf,mid=(a+b)/2;
    if(mid>=qa)
        ans=max(ans,query_max(a,mid,r<<1,qa,qb));
    if(qb>mid)
        ans=max(query_max(mid+1,b,r<<1|1,qa,qb),ans);
    return ans;
}

//查询区间最小值
ll query_min(ll a,ll b,ll r,ll qa,ll qb)
{
    if(a>=qa&&b<=qb)
        return tree_min[r];
    ll ans=inf,mid=(a+b)/2;
    if(mid>=qa)
        ans=min(ans,query_min(a,mid,r<<1,qa,qb));
    if(qb>mid)
        ans=min(ans,query_min(mid+1,b,r<<1|1,qa,qb));
    return ans;
}

int main()
{
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++) scanf("%lld",&A[i]);
    for(ll i=1;i<=n;i++)
    {
        scanf("%lld",&B[i]);
        sum[i]=sum[i-1]+B[i];
    }
    sum[0]=0;
    A[n+1]=-1e6-1;//为了将栈里的元素全部弹出
    build_max(0,n,1);//注意构造线段树下标为 0~n
    ll top=0,ans=-inf;;
    for(ll i=1;i<=n+1;i++)
    {
        while(top&&A[sta[top]]>A[i])
        {
            ll mid=sta[top],left,num=A[sta[top]];
            top--;
            left=sta[top];
            if(num>0) ans=max(ans,num*(query_max(0,n,1,mid,i-1)-query_min(0,n,1,left,mid-1)));
            else if(num<0) ans=max(ans,num*(query_min(0,n,1,mid,i-1)-query_max(0,n,1,left,mid-1)));
            else ans=max(ans,0);
        }
        sta[++top]=i;
    }
    printf("%lld",ans);
    return 0;
}