【ybtoj 高效进阶 4.4】【线段树】 求区间和
题目
解题思路
板子题
代码
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int n,m,w,x,y,l,r;
long long sum[400010];
void add(int k,int l,int r,int x,int y) //单点修改
{
if (l==r)
{
sum[k]+=y;
return;
}
int mid=(l+r)/2;
if (x<=mid)
add(k*2,l,mid,x,y);
else add(k*2+1,mid+1,r,x,y);
sum[k]=sum[k*2]+sum[k*2+1];
}
long long query(int k,int l,int r,int x,int y) //区间查询
{
if (x<=l&&r<=y) return sum[k]; //完全包括,直接返回
int mid=(l+r)/2;
long long da=0;
if (x<=mid) da+=query(k*2,l,mid,x,y);
if (y>mid) da+=query(k*2+1,mid+1,r,x,y); //二分下去
return da;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=m;i++)
{
scanf("%d",&w);
if (w==0)
{
scanf("%d%d",&x,&y);
add(1,1,n,x,y);
}
else {
scanf("%d%d",&l,&r);
printf("%lld\n",query(1,1,n,l,r));
}
}
return 0;
}