經過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!