经过3天的努力,终于解决了lazy_tag的一系列错误,终于可以发篇博客了~
lazy_tag是用来区间查询和修改的一种快速的方法,是线段树一种比较快速的操作。(虽然有些可以用树状数组来解决).
至于它的思想,可以去看刘汝佳的蓝书。
这里只贴代码~
/*
begin,end都是需要查询的区间。
*/
#define lt (node<<1)
#define rt (node<<1|1)
struct segment_tree{
int tree[N],lazy[N],a[N];
void create_tree(int node,int l,int r){
if(l==r){
tree[node]=a[l];
return ;
}
int mid=(l+r)>>1;
create_tree(lt,l,mid);
create_tree(rt,mid+1,r);
tree[node]=tree[lt]+tree[rt];
return ;
}
/*lazy_tag操作,每次查询到那个节点,就要把的信息传给自己的子节点*/
void down(int node,int l,int r){
if(!lazy[node]) return;
int mid=(l+r)>>1;
tree[lt]+=(mid-l+1)*lazy[node];
lazy[lt]+=lazy[node];
tree[rt]+=(r-mid)*lazy[node];
lazy[rt]+=lazy[node];
lazy[node]=0;
}
int query(int node,int l,int r,int begin,int end){
if(l>=begin && r<=end){
return tree[node];
}
down(node,l,r);//每次查到这,就得将信息向下传
int mid=(l+r)>>1,sb=0;
if(begin<=mid)sb+=query(lt,l,mid,begin,end);
if(end>mid)sb+=query(rt,mid+1,r,begin,end);
return sb;
}
void update(int node,int l,int r,int begin,int end,int add){
if(l>=begin && r<=end){
tree[node]+=(r-l+1)*add;
lazy[node]+=add;
return ;
}
int mid=(l+r)>>1;
if(begin<=mid) update(lt,l,mid,begin,end,add);
if(end>mid) update(rt,mid+1,r,begin,end,add);
tree[node]+=(min(r,end)-max(l,begin)+1)*add;
}
}d;
good luck to you!