Potentiometers ——线段树模板(单点更新,区间查询sum)

网址

坑点就是每输入一个cas都有一个空行。但最后一个没有。

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int n;
char ch[10];
int sum[maxn*4],A[maxn];
#define lson i*2, l, m
#define rson i*2+1, m+1, r
struct node
{
    int l,r,id;
}tree[maxn*4];
void build(int id,int l,int r)
{
    if(l==r)
    {
        sum[id]=A[l];
        return ;
    }
    int m=(l+r)/2;
    build(id*2,l,m);
    build(id*2+1,m+1,r);
    sum[id]=sum[id*2]+sum[id*2+1];
}
void Update(int point,int val,int id,int l,int r)
{
    if(l==r)
    {
        sum[id]=val;
        return ;
    }
    int m=(l+r)/2;
    if(point<=m)
        Update(point,val,id*2,l,m);
    else
        Update(point,val,id*2+1,m+1,r);
    sum[id]=sum[id*2]+sum[id*2+1];
}

int  query(int a,int b,int id,int l,int r)
{
    if(a<=l&&b>=r)
        return sum[id];
    int m=(l+r)/2;
    int res=0;
    if(a<=m)
        res+=query(a,b,id*2,l,m);
    if(b>m)
        res+=query(a,b,id*2+1,m+1,r);
    return res;
}


int main()
{
    int cas=0;
    int a,b;
    scanf("%d",&n);
   while(n!=0)
   {
       cas++;
        printf("Case %d:\n",cas);
       for(int i=1;i<=n;i++)
       {
           scanf("%d",&A[i]);
       }
       build(1,1,n);
       while(cin>>ch)
       {
       if(ch[0]=='M')
       {
           scanf("%d%d",&a,&b);
           cout<<query(a,b,1,1,n)<<endl;
       }
       else if(ch[0]=='S')
       {
           scanf("%d%d",&a,&b);
           Update(a,b,1,1,n);
       }
       else
       {
           scanf("%d",&n);
           if(n!=0)
           cout<<endl;
          break;
       }
       }
   }
   return 0;
}

猜你喜欢

转载自blog.csdn.net/xianpingping/article/details/80693125
今日推荐