线段树(模板)

#include<iostream>
#include<vector>
#include<cstdio>
#include<cmath>
using namespace std;
typedef long long ll;
int n,p,a,b,m,x,y,ans;
struct node
{
    int l,r,w,f;
} tree[400005];
void build(int k,int ll,int rr)//建树
{
    tree[k].l=ll;
    tree[k].r=rr;
    if(tree[k].l==tree[k].r)
    {
        scanf("%d",&tree[k].w);
        return ;
    }
    int m=(ll+rr)/2;
    build(k*2,ll,m);
    build(k*2+1,m+1,rr);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
void down(int k)//懒标记下传
{
    tree[k*2].f+=tree[k].f;
    tree[k*2+1].f+=tree[k].f;
    tree[k*2].w+=tree[k].f*(tree[k*2].r-tree[k*2].l+1);
    tree[k*2+1].w+=tree[k].f*(tree[k*2+1].r-tree[k*2+1].l+1);
    tree[k].f=0;
}
void point(int k)//单点查询
{
    if(tree[k].l==tree[k].r)
    {
        ans=tree[k].w;
        return ;
    }
    if(tree[k].f)
        down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m)
        point(k*2);
    else
        point(k*2+1);
}
void change_point(int k)//单点修改
{
    if(tree[k].l==tree[k].r)
    {
        tree[k].w+=y;
        return ;
    }
    if(tree[k].f)
        down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(x<=m)
        change_point(k*2);
    else
        change_point(k*2+1);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
void interval(int k)// 区间查询
{
    if(tree[k].l>=a&&tree[k].r<=b)
    {
        ans+=tree[k].w;
        return ;
    }
    if(tree[k].f)
        down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(a<=m)
        interval(k*2);
    if(b>m)
        interval(k*2+1);
}
void change_interval(int k)//区间修改
{
    if(tree[k].l>=a&&tree[k].r<=b)
    {
        tree[k].w+=(tree[k].r-tree[k].l+1)*y;
        tree[k].f+=y;
        return ;
    }
    if(tree[k].f)
        down(k);
    int m=(tree[k].l+tree[k].r)/2;
    if(a<=m)
        change_point(k*2);
    if(b>m)
        change_interval(k*2+1);
    tree[k].w=tree[k*2].w+tree[k*2+1].w;
}
int main()
{
    scanf("%d",&n);
    build(1,1,n);
    scanf("%d",&m);
    for(int i=1; i<=m; i++)
    {
        scanf("%d",&p);
        ans=0;
        if(p==1)
        {
            scanf("%d",&x);
            point(1);
            printf("%d\n",ans);
        }
        else if(p==2)
        {
            scanf("%d %d",&x,&y);
            change_point(1);
        }
        else if(p==3)
        {
            scanf("%d %d",&a,&b);
            interval(1);
            printf("%d\n",ans);
        }
        else
        {
            scanf("%d%d%d",&a,&b,&y);
            change_interval(1);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值