树状数组(模板)

Posted 2021-02-13 zut-syp

 

敌兵布阵

??题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=1166 ??
最简单的树状数组

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
using namespace std;

typedef long long LL;
const int Max_n=100005;
int d[Max_n],n;

void add(int x,int v){
    while(x<=n){
        d[x]+=v;
        x+=(x&-x);
    }
}

int quary(int x){
    int ans=0;
    while(x){
        ans+=d[x];
        x-=(x&-x);
    }
    return ans;
}

int main() {
    int t;
    scanf("%d",&t);
    for(int cnt=1;cnt<=t;cnt++){
        scanf("%d",&n);
        memset(d,0,sizeof(d));
        int x=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&x);
            add(i,x);
        }
        char opt[20];
        int l,r,ans=0;
        printf("Case %d:
",cnt);
        while(~scanf("%s",opt)&&opt[0]!=‘E‘){
            scanf("%d%d",&l,&r);
            if(opt[0]==‘Q‘){
                ans=quary(r)-quary(l-1);
                printf("%d
",ans);
            }else if(opt[0]==‘S‘){
                add(l,-r);
            }else{
                add(l,r);
            }
        }
    }
    return 0;
}

---