线段树封装板子

最新推荐文章于 2024-07-20 00:02:18 发布

余西子

最新推荐文章于 2024-07-20 00:02:18 发布

阅读量756

点赞数

本文链接：https://blog.csdn.net/qq_41955236/article/details/86651030

版权

本文深入探讨了段树数据结构的实现细节，包括初始化、更新、查询等核心操作，并通过一个具体的编程实例展示了如何使用段树解决区间更新和区间查询问题。文章提供了完整的代码示例，适合对数据结构和算法感兴趣的读者。

摘要由CSDN通过智能技术生成


#include<bits/stdc++.h>
#define mp make_pair
#define fi first
#define se second
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define eps 1e-8
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
const int MAXN=(int)1e5+5;
const int MOD=(int)1e9+7;
struct SegmentTree{
    #define lson (rt<<1)
    #define rson (rt<<1|1)
    #define mid  ((l+r)>>1)
    ll sum[1<<18|5],flag[1<<18|5];
    bool clr[1<<18|5];
    int n;
    void up(int l,int r,int rt,ll x){
        sum[rt]+=(r-l+1)*x;
        flag[rt]+=x;
    }
    void pushdown(int l,int r,int rt){
        if(clr[rt]==1){
            clr[lson]=clr[rson]=1;
            sum[lson]=sum[rson]=flag[lson]=flag[rson]=0;
            clr[rt]=0;
        }
        if(flag[rt]){
            up(l,mid,lson,flag[rt]);
            up(mid+1,r,rson,flag[rt]);
            flag[rt]=0;
        }
    }
    void pushup(int rt){
        sum[rt]=sum[lson]+sum[rson];
    }
    void build(int l,int r,int rt){
        flag[rt]=sum[rt]=clr[rt]=0;
        if(l==r)return;
        build(l,mid,lson);
        build(mid+1,r,rson);
        pushup(rt);
    }
    void init(int _n){
        n=_n;
        build(1,n,1);
    }
    void update(int l,int r,int rt,int x,int y,int z){
        if(x<=l&&r<=y){
            up(l,r,rt,z);
            return;
        }
        pushdown(l,r,rt);
        if(x<=mid)update(l,mid,lson,x,y,z);
        if(y>mid)update(mid+1,r,rson,x,y,z);
        pushup(rt);
    }
    int query(int l,int r,int rt,ll x){
        if(l==r){
            sum[rt]-=x;
            return l;
        }
        pushdown(l,r,rt);
        int re;
        if(sum[lson]>=x)re=query(l,mid,lson,x);
        else {
            re=query(mid+1,r,rson,x-sum[lson]);
            clr[lson]=1;
            sum[lson]=flag[lson]=0;
        }
        pushup(rt);
        return re;
    }
    int query(ll x){
        if(sum[1]<x)return 0;
        else return query(1,n,1,x);
    }
    void update(int l,int r,int x){
        update(1,n,1,l,r,x);
    }
}st;
int main()
{
int t;
scanf("%d",&t);
while(t--){
int n,m;
    scanf("%d%d",&n,&m);
    st.init(n);
    ll tot=0;
    while(m--){
        int op,l,r,x;
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d%d",&l,&r,&x);
            st.update(l,r,x);
        }
        else {
            scanf("%d",&x);
            int ans=st.query(x);
            if(ans==0)printf("Trote_w is sb\n");
            else {
                printf("%d\n",ans);
            }
        }
    }
}
    return 0;
}