洛谷P3374(线段树)(询问区间和,支持单点修改)

时间:2023-03-09 15:58:15
洛谷P3374(线段树)(询问区间和,支持单点修改)

洛谷P3374

 //询问区间和,支持单点修改
#include <cstdio> using namespace std; const int maxn=; struct treetype
{
int l,r,sum;
}; treetype a[*maxn];
int num[maxn]; void build(int k,int l,int r)
{
a[k].l=l;a[k].r=r;
if (l==r)
{
a[k].sum=num[l];
return;
}
int mid=(l+r)>>,i=k<<;
build(i,l,mid);
build(i+,mid+,r);
a[k].sum=a[i].sum+a[i+].sum;
}
void change(int k,int x,int t)
{
if (a[k].l==a[k].r)
{
a[k].sum+=t;
return;
}
int mid=(a[k].l+a[k].r)>>,i=k<<;
if (x<=mid) change(i,x,t);
else change(i+,x,t);
a[k].sum=a[i].sum+a[i+].sum;
}
int query(int k,int x,int y)
{
if (x<=a[k].l && a[k].r<=y) return a[k].sum;
int mid=(a[k].l+a[k].r)>>,ans=,i=k<<;
if (x<=mid) ans=query(i,x,y);
if (mid<y) ans+=query(i+,x,y);
return ans;
} int main()
{
int n,m;
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++) scanf("%d",&num[i]);
build(,,n);
//for (int i=1;i<=n*2;i++) printf("%d %d %d %d\n",i,a[i].l,a[i].r,a[i].sum);
for (int i=;i<=m;i++)
{
int t,x,y;
scanf("%d%d%d",&t,&x,&y);
if (t==) change(,x,y);
else printf("%d\n",query(,x,y));
}
return ;
}