【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,ri
typedef 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;
}