codevs 1080~1082 线段树练习系列(模板)

题目描述 Description
给你N个数,有两种操作:
1:给区间[a,b]的所有数增加X
2:询问区间[a,b]的数的和。

输入描述 Input Description
第一行一个正整数n,接下来n行n个整数,
再接下来一个正整数Q,每行表示操作的个数,
如果第一个数是1,后接3个正整数,
表示在区间[a,b]内每个数增加X,如果是2,
表示操作2询问区间[a,b]的和是多少。
pascal选手请不要使用readln读入

输出描述 Output Description
对于每个询问输出一行一个答案

样例输入 Sample Input
3
1
2
3
2
1 2 3 2
2 2 3

样例输出 Sample Output
9

数据范围及提示 Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000

题解:这三道题用同一份代码改一下main函数就能AC(这明明就是一道题好吧╭(╯^╰)╮),最简单的线段树模板。

代码如下:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
const ll maxn=233333;
int xl[maxn];
struct jgt
{///
    int l,r;
    ll sum,add;
}hah[maxn*4];
void build(int p,int l,int r)//建一棵线段树233 
{
    hah[p].l=l;
    hah[p].r=r;
    if(l==r)
    {
        hah[p].sum=xl[l];
        return ;
    }
    int mid=(l+r)/2;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);//线段树是一个二叉树233 
    hah[p].sum=hah[p*2].sum+hah[p*2+1].sum;
}
void spread(int p)//将标记下放
{
    if(hah[p].add)
    {
        hah[p*2].sum+=(hah[p*2].r-hah[p*2].l+1)*hah[p].add;
        hah[p*2+1].sum+=(hah[p*2+1].r-hah[p*2+1].l+1)*hah[p].add;
        hah[p*2].add+=hah[p].add;
        hah[p*2+1].add+=hah[p].add;
        hah[p].add=0;
    }
}
ll ask(int p,int l,int r)//区间查询 
{
    if(l<=hah[p].l&&hah[p].r<=r)
    {
        return hah[p].sum;
    }
    spread(p);//下放标记 
    ll ans=0;
    int mid=(hah[p].r+hah[p].l)/2;
    if(l<=mid) ans+=ask(p*2,l,r);
    if(mid<r) ans+=ask(p*2+1,l,r);
    return ans;
}
void change(int p,int l,int r,ll x)
//对于每一次修改我们做一个标记,只有需要的时候才进行计算 
{
    if(l<=hah[p].l&&hah[p].r<=r)
    {
        hah[p].sum+=(ll)(hah[p].r-hah[p].l+1)*x;
        hah[p].add+=x;
        return ;
    }
    spread(p);//下放标记 
    int mid=(hah[p].r+hah[p].l)/2;
    if(l<=mid) change(p*2,l,r,x);
    if(mid<r) change(p*2+1,l,r,x);//继续修改子节点 
    hah[p].sum=hah[p*2].sum+hah[p*2+1].sum;
}
int main()
{
    ll n,q;
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&xl[i]);
    build(1,1,n);//建树 
    scanf("%lld",&q);
    for(int i=1;i<=q;i++)
    {
        int ss;
        scanf("%d",&ss);
        if(ss==1)
        {
            int a,b;
            ll x;
            scanf("%d%d%lld",&a,&b,&x);
            change(1,a,b,x);//修改 
        }
        if(ss==2)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            printf("%lld\n",ask(1,a,b));//查询
        }
    }
    return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值