【代码记录】线段树的区间修改与区间查询

线段树是一种二叉搜索树，线段树将一段区间划分为若干个单位区间，每一个结点对应着原线段区间的一段，线段树维护的问题需要满足区间加法

 例题：https://www.luogu.com.cn/problem/P3372

线段树的建树

typedef long long ll;
void build(int p,int l,int r)
{
	tre[p].l=l;tre[p].r=r;
	if(tre[p].l==tre[p].r)
	 {//如果左右子树相等，就是叶子 
	 	tre[p].sum=a[l];
        return ;   
	 }
	int mid=l+r>>1;
	build(p*2,l,mid);
	//向左建树 
	build(p*2+1,mid+1,r);
	//向右建树 
	tre[p].sum=tre[p*2].sum+tre[p*2+1].sum;
}

区间修改区间查询和单点查询区间修改，区间查询单点查询的区别在于前者多了一个pushdown的操作，并不是两者代码的直接结合。而后者只是前者的一个子问题。pushdown的目的就是让修改区间的懒标记不断向下传递，同时让自己的叶子结点加上k*（r-l）(k是要求修改的值).然后自己清空

pushdown的操做

void pushdown(int p)
{//向下传递懒标记 
	if(tre[p].lazy)
    {
    	tre[p*2].sum+=tre[p].lazy*(tre[p*2].r-tre[p*2].l+1);
    	tre[p*2+1].sum+=tre[p].lazy*(tre[p*2+1].r-tre[p*2+1].l+1);
    	//让自己的子节点加上k*(r-l) 
    	tre[p*2].lazy+=tre[p].lazy;
    	tre[p*2+1].lazy+=tre[p].lazy;
    	tre[p].lazy=0;
    	//传递完后自己清零 
	}
}

线段树的修改操作

区间修改的时候，我们按照如下原则：

1、如果当前区间被完全覆盖在目标区间里，将这个区间的sum+k*(tree[i].r-tree[i].l+1)

2、如果没有完全覆盖，则先下传懒标记

3、如果这个区间的左儿子和目标区间有交集，那么搜索左儿子

4、如果这个区间的右儿子和目标区间有交集，那么搜索右儿子

void change(int p,int l,int r,int k)
{//区间修改 
	if(tre[p].l>=l&&tre[p].r<=r)
	{//如果当前区间完全包含在修改区间内 
		tre[p].sum+=(ll)k*(tre[p].r-tre[p].l+1);
		tre[p].lazy+=k;
		return ;
	}
	pushdown(p);
	int mid=tre[p].l+tre[p].r>>1;
	if(l<=mid) change(p*2,l,r,k);
	if(r>mid)  change(p*2+1,l,r,k);
	tre[p].sum=tre[p*2].sum+tre[p*2+1].sum;
}

区间查询

ll search(int p,int l,int r)
{//区间查询 
	if(tre[p].l>=l&&tre[p].r<=r)
	  return tre[p].sum;
	pushdown(p);
	int mid=tre[p].l+tre[p].r>>1;
	long long ans(0);
	if(l<=mid) ans+=search(p*2,l,r);
	if(r>mid) ans+=search(p*2+1,l,r);
	return ans;
}