HDU1166——树状数组

最新推荐文章于 2020-06-09 19:27:52 发布

mlm5678

最新推荐文章于 2020-06-09 19:27:52 发布

阅读量148

点赞数

分类专栏： ac 树状数组

本文链接：https://blog.csdn.net/mlm5678/article/details/89078759

版权

ac 同时被 2 个专栏收录

95 篇文章 0 订阅

订阅专栏

树状数组

2 篇文章 0 订阅

订阅专栏

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
int c[100005],a[100005];
int n;
int lowbit(int x){///清空高位1，计算2^k
    return x&(-x);
}
void  modify(int x,int add){///更新
    while(x<=n){///n为数组总长度
        c[x]+=add;
        x+=lowbit(x);
    }
}
int sum(int x){///求和，从1到x的和
    int ans=0;
    while(x>0){///大于0
        ans+=c[x];
        x-=lowbit(x);
    }
    return ans;
}
int get(int x,int y){
    return sum(y)-sum(x-1);
}
int main(){
    int i,j,T;
    int x,y;
    scanf("%d",&T);
    int k=0;
    while(T--){
        k++;
        memset(a,0,sizeof(a));
        memset(c,0,sizeof(c));
        scanf("%d",&n);
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
            modify(i,a[i]);
        }
        printf("Case %d:\n",k);
        char s[10];
        while(scanf("%s",s)==1){
            if(strcmp(s,"End")==0)
                break;
            scanf("%d %d",&x,&y);
            if(strcmp(s,"Query")==0){
                printf("%d\n",get(x,y));
            }
            if(strcmp(s,"Add")==0){
                a[x]+=y;///更新后再对树状数组更新
                modify(x,y);
            }
            if(strcmp(s,"Sub")==0){
                a[x]-=y;
                modify(x,-y);
            }
        }
    }
    return 0;
}