vscode 动态最大子段和

题目链接:戳我

对于最大子段和,我们只需要维护四个变量——maxl,maxr,maxs,sum(分别表示区间最大前缀子段和,区间最大后缀子段和,区间最大子段和,区间所有数的和)

然后合并的时候是这样的:

t[x].sum=t[ls(x)].sum+t[rs(x)].sum;
t[x].maxs=max(max(t[ls(x)].maxs,t[rs(x)].maxs),t[ls(x)].maxr+t[rs(x)].maxl);
t[x].maxl=max(t[ls(x)].maxl,t[ls(x)].sum+t[rs(x)].maxl);
t[x].maxr=max(t[rs(x)].maxr,t[ls(x)].maxr+t[rs(x)].sum);

然后想要强调的是,询问的时候一定不能像普通线段树那样写,因为我们需要考虑三种情况。一种是该节点表示区间的答案应该从左子节点表示区间取得,一种是该节点表示区间的答案是应该从右子节点表示区间取得,一种是该节点表示区间的答案应该从左右节点中合并取得。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define MAXN 2000010
using namespace std;
int n,m;
long long a[MAXN];
struct Node{long long maxs,maxl,maxr,sum;}t[MAXN<<2];
inline int ls(int x){return x<<1;}
inline int rs(int x){return x<<1|1;}
inline void push_up(int x)
{
    t[x].sum=t[ls(x)].sum+t[rs(x)].sum;
    t[x].maxs=max(max(t[ls(x)].maxs,t[rs(x)].maxs),t[ls(x)].maxr+t[rs(x)].maxl);
    t[x].maxl=max(t[ls(x)].maxl,t[ls(x)].sum+t[rs(x)].maxl);
    t[x].maxr=max(t[rs(x)].maxr,t[ls(x)].maxr+t[rs(x)].sum);
}
inline void build(int x,int l,int r)
{
    if(l==r) 
    {
        t[x].sum=t[x].maxs=t[x].maxl=t[x].maxr=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(ls(x),l,mid);
    build(rs(x),mid+1,r);
    push_up(x);
    //printf("x=%d l=%d r=%d maxs=%d maxl=%d maxr=%d sum=%d\n",x,l,r,t[x].maxs,t[x].maxl,t[x].maxr,t[x].sum);
}
inline Node query(int x,int l,int r,int ll,int rr)
{
    //printf("x=%d l=%d r=%d ll=%d rr=%d\n",x,l,r,ll,rr);
    if(ll==l&&r==rr) {return t[x];}
    int mid=(l+r)>>1;
    if(rr<=mid) return query(ls(x),l,mid,ll,rr);
    else if(mid<ll) return query(rs(x),mid+1,r,ll,rr);
    else
    {
        Node lson=query(ls(x),l,mid,ll,mid),rson=query(rs(x),mid+1,r,mid+1,rr),ans;
        ans.sum=lson.sum+rson.sum;
        ans.maxs=max(max(lson.maxs,rson.maxs),lson.maxr+rson.maxl);
        ans.maxl=max(lson.maxl,lson.sum+rson.maxl);
        ans.maxr=max(rson.maxr,lson.maxr+rson.sum);
        return ans;
    }
}

int main()
{
    freopen("ce1.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        long long l,r;
        scanf("%lld%lld",&l,&r);
        printf("%lld\n",query(1,1,n,l,r).maxs);
    }
    return 0;
}
上一篇:第八周作业


下一篇:java – 在同一版本中执行JUnit 4和JUnit 5测试