首页 > 代码库 > 线段树模板
线段树模板
自己写的模板,方便以后查看
求最大值的:
#include<iostream> #include<cstdio> using namespace std; #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 const int maxn=222222; int value[maxn<<2]; void PushUp(int rt) { value[rt]=max(value[rt<<1],value[rt<<1|1]); } void btree(int l,int r,int rt) { //初始为(1,n,1) if(l==r) { scanf("%d",&value[rt]); return ; } int m=(l+r)>>1; btree(ls); btree(rs); PushUp(rt); } void update(int x,int v,int l,int r,int rt) { //初始化为(a,b,1,n,1) if(l==r) { value[rt]=v; return ; } int m=(l+r)>>1; if(x<=m)update(x,v,ls); else update(x,v,rs); PushUp(rt); } int query(int L,int R,int l,int r,int rt) { //初始化为(a,b,1,n,1) if(L<=l&&r<=R)return value[rt]; int m=(l+r)>>1,ans=0; if(L<=m)ans=max(ans,query(L,R,ls)); if(R>m)ans=max(ans,query(L,R,rs)); return ans; } int main() { }
求区间和:
#include<iostream> #include<cstdio> using namespace std; #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 const int maxn=50005; int value[maxn<<2],ans; void PushUp(int rt) { value[rt]=value[rt<<1]+value[rt<<1|1]; } void btree(int l,int r,int rt) { //初始为(1,n,1) if(l==r) { scanf("%d",&value[rt]); return ; } int m=(l+r)>>1; btree(ls); btree(rs); PushUp(rt); } void update(int x,int v,int l,int r,int rt) { //初始化为(a,b,1,n,1) if(l==r) { value[rt]+=v; return ; } int m=(l+r)>>1; if(x<=m)update(x,v,ls); else update(x,v,rs); PushUp(rt); } void query(int L,int R,int l,int r,int rt) { //初始化为(a,b,1,n,1) if(L<=l&&r<=R) { ans+=value[rt]; return ; } if(l==r)return; int m=(l+r)>>1; if(R<=m)query(L,R,ls); else if(L>m)query(L,R,rs); else { query(L,R,ls); query(L,R,rs); } } int main() { return 0; }
区间更新的:
#include<iostream> #include<cstdio> using namespace std; #define ll long long #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 const int maxn=100005; ll value[maxn<<2],add[maxn<<2]; void PushUp(int rt) { value[rt]=value[rt<<1]+value[rt<<1|1]; } void PushDown(int rt,int m)//m是当前节点区间的范围 { //例如1-5 右儿子4-5,左儿子1-3 m=5,m>>1=2刚好是右儿子区间范围,m-(m>>1)=3为左儿子区间范围 if(add[rt])//向下更新子节点 { add[rt<<1]+=add[rt];//更新左儿子的lazy标记 add[rt<<1|1]+=add[rt];//右儿子 value[rt<<1]+=(m-(m>>1))*add[rt];//更新左儿子的权值,应该乘上左儿子区间(r-l+1) value[rt<<1|1]+=(m>>1)*add[rt];//应该乘右儿子区间范围 add[rt]=0; } } void btree(int l,int r,int rt) { //初始化1,n,1 add[rt]=0; if(l==r) { scanf("%lld",&value[rt]); return ; } int m=(l+r)>>1; btree(ls); btree(rs); PushUp(rt); } void update(int L,int R,int c,int l,int r,int rt) { //初始化a,b,c,1,n,1 if(L<=l&&r<=R)//给定区间大于现有区间 { add[rt]+=c;//标记加上 value[rt]+=(ll)c*(r-l+1);//更新当前节点 return ; } PushDown(rt,r-l+1);//向下更新子节点因为下一步要使用子节点 int m=(l+r)>>1; if(L<=m)update(L,R,c,ls); if(R>m)update(L,R,c,rs); PushUp(rt);//向上关系父节点 } ll query(int L,int R,int l,int r,int rt) { //初始化为a,b,1,n,1 if(L<=l&&r<=R)return value[rt]; PushDown(rt,r-l+1); int m=(l+r)>>1; ll ans=0; if(L<=m)ans+=query(L,R,ls); if(R>m)ans+=query(L,R,rs); return ans; } int main() { }
线段树模板
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。