不太强的线段树区间更新模板

由于每次比赛线段树都是手写的,风险较大易背锅,为了保证线段树不再翻车,这一次认认真真的写了一个线段树模板。

#include
#include
#include
#include
#include
using namespace std;
#define lson k<<1
#define rson k<<1|1
const int maxn=1e5+5;
struct node
{
    long long sum,lazy;
    long long mmin,mmax;
    int siz;
}tree[maxn<<2];
long long a[maxn];
void pushdown(int k)
{
    if(tree[k].lazy)
    {
        tree[lson].sum+=tree[k].lazy*tree[lson].siz;
        tree[rson].sum+=tree[k].lazy*tree[rson].siz;
        tree[lson].mmin+=tree[k].lazy;
        tree[rson].mmin+=tree[k].lazy;
        tree[lson].mmax+=tree[k].lazy;
        tree[rson].mmax+=tree[k].lazy;
        tree[lson].lazy+=tree[k].lazy;
        tree[rson].lazy+=tree[k].lazy;
        tree[k].lazy=0;
    }
}
void pushup(int k)
{
    tree[k].sum=tree[lson].sum+tree[rson].sum;
    tree[k].mmin=min(tree[lson].mmin,tree[rson].mmin);
    tree[k].mmax=max(tree[lson].mmax,tree[rson].mmax);
}
void build(int l,int r,int k)
{
    tree[k].siz=r-l+1;
    tree[k].lazy=0;
    if(l==r)
    {
        tree[k].sum=tree[k].mmin=tree[k].mmax=a[l];
        return ;
    }
    int m=((l+r)>>1);
    build(l,m,k<<1);
    build(m+1,r,k<<1|1);
    pushup(k);
}
void update(int p,int c,int l,int r,int k)
{
    if(l==r)
    {
        tree[k].sum+=c;
        tree[k].mmax+=c;
        tree[k].mmin+=c;
        return ;
    }
    pushdown(k);
    int mid=((l+r)>>1);
    if(p<=mid) update(p,c,l,mid,lson);
    else update(p,c,mid+1,r,rson);
    pushup(k);
}
void qjupdate(int L,int R,int c,int l,int r,int k)
{
    if(L<=l&&r<=R)
    {
        tree[k].sum+=c*tree[k].siz;
        tree[k].mmax+=c;
        tree[k].mmin+=c;
        tree[k].lazy+=c;
        return ;
    }
    pushdown(k);
    int mid=((l+r)>>1);
    if(L<=mid) qjupdate(L,R,c,l,mid,lson);
    if(R>mid) qjupdate(L,R,c,mid+1,r,rson);
    pushup(k);
}
long long query(int L,int R,int l,int r,int k)
{
    if(L<=l&&r<=R)
    {
        return tree[k].???;
    }
    pushdown(k);
    int mid=((l+r)>>1);
    long long ans=0,mmin=inf,mmax=0;
    if(L<=mid)
    {
        ans+=query(L,R,l,mid,lson);
        mmin=min(mmin,query(L,R,l,mid,lson));
        mmax=max(mmax,query(L,R,l,mid,lson));
    }
    if(R>mid)
    {
        ans+=query(L,R,mid+1,r,rson);
        mmin=min(mmin,query(L,R,mid+1,r,rson))
        mmax=max(mmax,query(L,R,mid+1,r,rson));
    }
    return ???;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    build(1,n,1);
    ???
    return 0;
}

 

你可能感兴趣的:(acm)