题目描述 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;
}