【NYOJ 116】士兵杀敌（二） 【树状数组 or 线段树 】

士兵杀敌（二）
时间限制：1000 ms | 内存限制：65535 KB
难度：5
描述
南将军手下有N个士兵，分别编号1到N，这些士兵的杀敌数都是已知的。

小工是南将军手下的军师，南将军经常想知道第m号到第n号士兵的总杀敌数，请你帮助小工来回答南将军吧。

南将军的某次询问之后士兵i可能又杀敌q人，之后南将军再询问的时候，需要考虑到新增的杀敌数。

输入
只有一组测试数据
第一行是两个整数N,M，其中N表示士兵的个数( 1 < N < 1000000)，M表示指令的条数。(1 < M <100000)
随后的一行是N个整数，ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行是一条指令，这条指令包含了一个字符串和两个整数，首先是一个字符串，如果是字符串QUERY则表示南将军进行了查询操作，后面的两个整数m,n，表示查询的起始与终止士兵编号；如果是字符串ADD则后面跟的两个整数I,A(1<=I<=N,1<=A<=100),表示第I个士兵新增杀敌数为A.
输出
对于每次查询，输出一个整数R表示第m号士兵到第n号士兵的总杀敌数，每组输出占一行
样例输入
5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5
样例输出
6
8
8
20
来源

线段树代码

#include<bits/stdc++.h>using namespace std ;#define ll o<<1#define rr o<<1|1#define lson o<<1,le,mid#define rson o<<1|1,mid+1,ritypedef long long LL ;const int MAXN =  1000000+100;const int MAXM = 1e5 ;const int mod  = 1007 ;const int inf  = 0x3f3f3f3f;struct Tree{    int l,r,sum;}tree[MAXN<<2];void up(int o){    tree[o].sum=tree[ll].sum+tree[rr].sum;}void build(int o,int le ,int ri){    tree[o]={le,ri,0};    if(le==ri){         scanf("%d",&tree[o].sum);         return ;    }    int mid=(le+ri)>>1;    build(lson);    build(rson);    up(o);}void update(int o,int pos,int val){    if(tree[o].l==tree[o].r&&tree[o].r==pos) {        tree[o].sum+=val;        return ;    }    int mid=(tree[o].l+tree[o].r)>>1;    if(pos>mid) update(rr,pos,val);    else update(ll,pos,val);    up(o);}int query(int o,int le,int ri){    if(tree[o].l>=le&&tree[o].r<=ri) return tree[o].sum;    int mid=(tree[o].l+tree[o].r)>>1;    if(le>mid) return query(rr,le,ri);    else if(ri<=mid) return query(ll,le,ri);    else return query(ll,le,mid)+query(rr,mid+1,ri);}int main(){     int n,m;     scanf("%d%d",&n,&m);     build(1,1,n);     while(m--){        char op[10];        scanf("%s",op);        if(strcmp(op,"QUERY")==0) {            int a,b;            scanf("%d%d",&a,&b);            printf("%d\n",query(1,a,b));          }else {            int a,b;            scanf("%d%d",&a,&b);            update(1,a,b);          }      }    return  0;}

树状数组代码

#include<cstdio>#include<cstring>using namespace std;#define lowbit(x) (x&(-x))const int MAXN =1000000+10;const int MAXM =1e5;const int mod  =1e9+7;int n,m;int arr[MAXN];void add(int x,int y){    while(x<=n){        arr[x]+=y;        x+=lowbit(x);    }}int query(int x){    int sum=0;    while(x>0){        sum+=arr[x];        x-=lowbit(x);    }    return sum;}int main(){    scanf("%d%d",&n,&m);    memset(arr,0,sizeof(arr));    for(int i=1;i<=n;i++) {        int z;        scanf("%d",&z);        add(i,z);    }    char op[10];int a,b;    while(m--){    scanf("%s%d%d",op,&a,&b);        if(strcmp(op,"ADD")==0) add(a,b);        else printf("%d\n",query(b)-query(a-1));         }    return 0;}