• 线段树各类操作



    #include <iostream> #include <bits/stdc++.h> #define maxn 800005 using namespace std; typedef long long ll; ll tree[maxn]={0}; ll lazy[maxn]={0}; ll a[200005]={0}; int n; void pushup(int rt) { tree[rt]=max(tree[rt*2],tree[rt*2+1]); } void buildtree(int l,int r,int rt) { if(l==r) { tree[rt]=a[l]; return; } int mid=(l+r)/2; buildtree(l,mid,rt*2); buildtree(mid+1,r,rt*2+1); pushup(rt); } void pushdown(int rt) { if(lazy[rt]) { lazy[rt*2]+=lazy[rt]; lazy[rt*2+1]+=lazy[rt]; tree[rt*2]+=lazy[rt]; tree[rt*2+1]+=lazy[rt]; lazy[rt]=0; } } void add(int l,int r,int L,int R,int rt,int val) { if(L<=l&&R>=r) { lazy[rt]+=val; tree[rt]+=val; return; } if(lazy[rt]) pushdown(rt); int mid=(l+r)/2; if(L<=mid) add(l,mid,L,R,rt*2,val); if(R>mid) add(mid+1,r,L,R,rt*2+1,val); pushup(rt); } ll query(int l,int r,int L,int R,int rt) { if(L<=l&&R>=r) { return tree[rt]; } int mid=(l+r)/2; pushdown(rt); ll ret=1e9; if(L<=mid) ret=query(l,mid,L,R,rt*2); if(R>mid) ret=max(ret,query(mid+1,r,L,R,rt*2+1)); return ret; } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); } buildtree(1,n,1); int m; scanf("%d",&m); for(int i=1;i<=m;i++) { char c; scanf("%c",&c); if(c=='a') { int x,y,val; scanf("%d%d%d",&x,&y,&val); add(1,n,x,y,1,val); } else if(c=='q') { int x,y; scanf("%d%d",&x,&y); cout<<query(1,n,x,y,1)<<endl; } } return 0; }

      RMQ操作。

    #include <bits/stdc++.h>
    typedef long long ll;
    const int MAXN=401000;
    using namespace std;
    ll mx[MAXN]={0};
    ll a[MAXN]={0};
    ll lazy[MAXN]={0};
    void pushup(int rt)
    {
        mx[rt]=mx[rt*2]+mx[rt*2+1];
    }
    void buildtree(int l,int r,int rt)
    {  lazy[rt]=0;
        if(l==r)
        {
            mx[rt]=a[l];
            return;
        }
        int mid=(l+r)/2;
        buildtree(l,mid,rt*2);
        buildtree(mid+1,r,rt*2+1);
        pushup(rt);
    }
    void pushdown(int rt,int le,int ri)
    {
        if(lazy[rt])
        {
            lazy[rt*2]+=lazy[rt];
            lazy[rt*2+1]+=lazy[rt];
            mx[rt*2]+=lazy[rt]*le;
            mx[rt*2+1]+=lazy[rt]*ri;
            lazy[rt]=0;
        }
    }
    void update(int L,int R,int val,int l,int r,int rt)
    {
        if(L<=l&&r<=R)
        {
            mx[rt]+=1ll*val*(r-l+1);
            lazy[rt]+=val;
            return ;
        }
        int mid=(l+r)/2;
        pushdown(rt,mid-l+1,r-mid);
        if(L<=mid) update(L,R,val,l,mid,rt*2);
        if(R>mid) update(L,R,val,mid+1,r,rt*2+1);
        pushup(rt);
    }
    ll query(int L,int R,int l,int r,int rt)
    {
        if(L<=l&&R>=r)
        {
            return mx[rt];
        }
        int mid=(l+r)/2;
        pushdown(rt,mid-l+1,r-mid);
        ll ret=0;
        if(L<=mid) ret=ret+query(L,R,l,mid,rt*2);
        if(R>mid) ret=ret+query(L,R,mid+1,r,rt*2+1);
        return ret;
    }
    int main()
    {
        int n,m;
        ios::sync_with_stdio(false);
        cin>>n>>m;
        for(int i=1;i<=n;++i)
        {
            //scanf("%lld",&a[i]);
                    cin>>a[i];
        }
        buildtree(1,n,1);
        for(int i=1;i<=m;++i)
        {
            char c;
            int x,y;
            cin>>c>>x>>y;
            if(c=='Q')
            {
                cout<<query(x,y,1,n,1)<<endl;
            }
            else
            {
                int val;
                cin>>val;
                update(x,y,val,1,n,1);
            }
        }
        return 0;
    }
    

      

      区间修改/求和。

  • 相关阅读:
    2021.5.16 Android聊天功能
    2021.5.15 Android Gestures示例
    2021.5.14 程序员修炼之路:从小工到专家阅读笔记02
    KL 散度和交叉熵
    UBOOT学习
    UCOSII学习
    cortex-M3/M4体系学习
    一步步写RTOS
    38 操作系统-中断处理与特权级转移
    MDK、IAR将变量定义到指定位置
  • 原文地址:https://www.cnblogs.com/zyf3855923/p/9069372.html
Copyright © 2020-2023  润新知