线段树 模版

  #include <iostream>
#include<stdio.h>
using namespace std;

const int maxn=100002;
int sum[maxn*4]={0};//这个要初始化为0
int col[maxn*4]={0};//标记每个节点,初始化为0
//将子节点的数据传到根节点
void PushUp(int rt)
{//存放节点的总和
    sum[rt]=sum[rt*2]+sum[rt*2+1];
}
//将m个值做标记
void PushDownAdd(int rt,int m)
{
    //如果该点做了标记
    if(col[rt])
    {
        //那么为该点的孩子节点也做标记
		col[rt*2]+=col[rt];
		col[rt*2+1]+=col[rt];
        //并修改该点孩子节点的值
        sum[rt*2]+=(m-m/2)*col[rt];
        sum[rt*2+1]+=(m/2)*col[rt];
        col[rt]=0;//取消该点的标记
    }
}

//将m个值做标记
void PushDownValue(int rt,int m)
{
    //如果该点做了标记
    if(col[rt])
    {
        //那么为该点的孩子节点也做标记
        col[rt*2]=col[rt*2+1]=col[rt];
        //并修改该点孩子节点的值
        sum[rt*2]=(m-m/2)*col[rt];
        sum[rt*2+1]=(m/2)*col[rt];
        col[rt]=0;//取消该点的标记
    }
}

//递归创建树,并且更新到根节点,sum保存了这个线段的值
void build(int l,int r,int rt)
{
    if(l==r)
    {
		cin>>sum[rt];
        return;
    }
    int m=(l+r)/2;
    build(l,m,rt*2);
    build(m+1,r,rt*2+1);
    PushUp(rt);
}

//在l到r区间中查找left到right区间的和
int query(int left,int right,int l,int r,int rt)
{

    if(l==left && r==right)
    {
        return sum[rt];
    }
	PushDownAdd(rt,r-l+1);
    int mid=(r+l)/2;
    int ret=0;
    if(right<=mid)//在左孩子
	{
        ret+=query(left,right,l,mid,rt*2);
	}
    else if(left>=mid+1)//在右孩子
        ret+=query(left,right,mid+1,r,rt*2+1);
    else//一半在左孩子一半在右孩子
    {
        //先查找left到mid区间的和,在计算mid+1到right区间的和
        ret+=query(left,mid,l,mid,rt*2);
        ret+=query(mid+1,right,mid+1,r,rt*2+1);
    }
    return ret;
}
//在l到r区间中更新left到right区间的值加上add,并且更新它父亲的值
void updateAdd(int left,int right,int add,int l,int r,int rt)
{
    //要更新的这个区间在l到r区间内
    if(left<=l && r<=right)
    {//只在该点做修改,其他点就不在修改,只是做标记
        col[rt]+=add;//在该点做标记---标记累加
        sum[rt]+=add*(r-l+1);//该点的值做修改
        return;
    }
    //修改该点的孩子的值
    PushDownAdd(rt,r-l+1);
    int m=(l+r)/2;
    if(left<=m)
        updateAdd(left,right,add,l,m,rt*2);
    if(right>m)
        updateAdd(left,right,add,m+1,r,rt*2+1);
    PushUp(rt);//更改rt节点的总和
}


//在l到r区间中更新left到right区间的值为add,并且更新它父亲的值
void updateValue(int left,int right,int value,int l,int r,int rt)
{
    //要更新的这个区间在l到r区间内
    if(left<=l && r<=right)
    {//只在该点做修改,其他点就不在修改,只是做标记
        col[rt]=value;//在该点做标记
        sum[rt]=value*(r-l+1);//该点的值做修改
        return;
    }
    //修改该点的孩子的值
    PushDownValue(rt,r-l+1);
    int m=(l+r)/2;
    if(left<=m)
        updateValue(left,right,value,l,m,rt*2);
    if(right>m)
        updateValue(left,right,value,m+1,r,rt*2+1);
    PushUp(rt);//更改rt节点的总和
}

int main()
{
	int N,Q;
    char ch;
    int a,b;
	int c;
    scanf("%d%d",&N,&Q);
	build(1,N,1);
    while(Q--)
    {
		cin>>ch;
        if(ch=='Q')
        {
            scanf("%d%d",&a,&b);
            printf("%d\n",query(a,b,1,N,1));
        }
        else if(ch=='C')
        {
            scanf("%d%d%d",&a,&b,&c);
            updateAdd(a,b,c,1,N,1);
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

疯的世界

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值