线段树之单点修改(斑点蛇上数斑点)



#include<iostream>
#include<string>
using namespace std;
/*刚学线段树,调了好一会。
*单点修改的模板。
*/
const int Kmax=50000+20;
int sq[Kmax<<2];

void up(int p){
    sq[p]=sq[p<<1]+sq[(p<<1)+1];
}

void modify(int p,int l,int r,int i,int ai){
    if(l==r){
        sq[p]+=ai;
        //  cout<<i<<":"<<sq[p]<<" ";
        return;
    }
    int mid=(l+r)/2;
    if(i<=mid){
        modify(p<<1,l,mid,i,ai);
    }
    else{
        modify((p<<1)+1,mid+1,r,i,ai);
    }
    up(p);
}
int query(int p,int l,int r,int a,int b){
    if(a<=l&&r<=b){
       // cout<<l<<"-"<<r<<":"<<sq[p]<<" $$ ";
        return sq[p];

    }
    int mid=(l+r)/2,ans=0;
    if(mid>=a){
        ans+=query(p<<1,l,mid,a,b);
    }
    if(mid<b){
        ans+=query((p<<1)+1,mid+1,r,a,b);
    }
    return ans;
}

int main(){
    std::ios::sync_with_stdio(false);
    int n;
    cin>>n;
    int ai;
    for(int i=1;i<=n;i++){
        cin>>ai;
        modify(1,1,n,i,ai);
    }
   // cout<<sq[1]<<endl;
    string op;
    int a,b;
    while (cin>>op) {
       if(op[0]=='E') break;
       if(op[0]=='A'){
           cin>>a>>b;
           modify(1,1,n,a,b);
       }
       else if(op[0]=='S'){
           cin>>a>>b;
           modify(1,1,n,a,-b);
       }
       else{
           cin>>a>>b;
           cout<<query(1,1,n,a,b)<<endl;
       }
    }
    return 0;
}

阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页