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;
}