求助线段树!
查看原帖
求助线段树!
168334
封禁用户楼主2021/8/2 19:40

RT,不知道哪里出了问题,反复看了几个线段树题解也找不出。

#include<bits/stdc++.h>
using namespace std;
const int maxn=500001+10;
const int INF=0x3f3f3f3f;
int n,m;
int a[maxn],tree[4*maxn];
void built(int l,int r,int rt)
{
  if(l==r)  {tree[rt]=a[l];return;}
  int mid=(l+r)>>1;
  built(l,mid,rt<<1);
  built(mid+1,r,rt<<1|1);
  tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
int query(int l,int r,int rt,int xl,int xr)
{
  if(l==xl && r==xr)  return tree[rt];
  int mid=(l+r)>>1,res=0;
  if(xr<=mid)  res+=query(l,mid,rt<<1,xl,xr);
  else
    if(xl>mid)  res+=query(mid+1,r,rt<<1|1,xl,xr);
    else  res=res+query(l,mid,rt<<1,xl,mid)+query(mid+1,r,rt<<1|1,mid+1,xr);
  return res;
}
void change(int l,int r,int rt,int pos,int val)
{
  if(l==r)  {tree[rt]+=val;return;}
  int mid=(l+r)>>1;
  if(pos<=mid)  return change(l,mid,rt<<1,pos,val);
  else  return change(mid+1,r,rt<<1|1,pos,val);
  tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}

int main()
{
	scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
      scanf("%d",&a[i]);
	built(1,n,1);
	while(m--)
    {
      int p,b,c;
      scanf("%d%d%d",&p,&b,&c);
      if(p==1)  change(1,n,1,b,c);
      else  printf("%d\n",query(1,n,1,b,c));
    }
	return 0;
}

2021/8/2 19:40
加载中...