TOJ 3198 单点更改 区间查询

3198: 区间和
时间限制(普通/Java):1000MS/3000MS 内存限制:65536KByte
总提交: 967 测试通过:302
描述

给定n个数据，有两个操作，加减其中的一个数据，当然还可查询在某段数据的和。

输入

输入数据有多组，每组数据的
第一行输入n，1=<n<=500000,代表数据的个数。
第二行输入具体数据，数据为正整数，范围在1到10000.
第三行输入m，1<=m<=100000,表示操作的次数。包含了修改和查询操作。
下面m行就是具体的操作了。
C i x 表示为第i个元素加上x，x范围在1到10000.
Q i j 表示查询区段i到j的和。保证输入的i<=j.
以EOF结束。

输出

输出查询后的区段和。

样例输入

8
1 5 9 11 2 8 15 6
4
Q 1 3
C 2 10
Q 1 4
Q 2 5

样例输出

15
36
37

提示：类型最好定义为__int64
题解：~~~~~~~线段数，单点修改和区间查询，父节点记录和就好了，用一个数组去存树，一般开4倍

#include<cstdio>
#include<iostream>
#include<string.h>
using namespace std;
const int maxn=500005;
int arr[maxn],tree[maxn*4]; 
void build(int node,int l,int r){ //建树
    if(l==r)
        tree[node] = arr[l]; // 走到最深的节点，赋值
    else{
        int mid = (l+r)/2;
        int left_node = 2*node + 1;	//左儿子
        int right_node = 2*node + 2;	//右儿子

        build(left_node,l,mid);
        build(right_node,mid+1,r); // 两边递归
        tree[node]=tree[left_node]+tree[right_node]; // 回溯父节点记录和
    }
}
void update(int node,int l,int r,int idx,int val){
    if(l==r)
        arr[idx]+=val,tree[node]+=val;//找到点
    else{
        int mid = (l+r)/2;
        int left_node = 2*node + 1;
        int right_node = 2*node + 2;

        if(idx<=mid&&idx>=l) //比中点小 左边搜
            update(left_node,l,mid,idx,val);
        else
            update(right_node,mid+1,r,idx,val);
        tree[node]=tree[left_node]+tree[right_node]; //修改完回溯
    }												//修改父节点的和
}
int query(int node,int l,int r,int L,int R){
    if(r<L || l>R)	//越界不走，优化
        return 0;
    else if ((l>=L && r<=R) || l==r) //区间内直接返回
    	 return tree[node];
    else{
        int mid = (l+r)/2;
        int left_node = 2*node + 1;
        int right_node = 2*node + 2;
        int query_left=query(left_node,l,mid,L,R); //左边搜
        int query_right=query(right_node,mid+1,r,L,R);//右边搜
        return query_left+query_right;	//返回和
    }
}
int main(){
    int n;
    while(~scanf("%d",&n)){
        memset(tree,0,sizeof tree);
        for(int i=0;i<n;i++)
            scanf("%d",&arr[i]);
        build(0,0,n-1);
        int m;
        scanf("%d",&m);
        while(m--){
            getchar();
            char c;
            scanf("%c",&c);
            if(c=='C'){
                int idx,val;
                scanf("%d%d",&idx,&val);
                update(0,0,n-1,idx-1,val);
            }
            else{
                int L,R;
                scanf("%d%d",&L,&R);
                printf("%d\n",query(0,0,n-1,L-1,R-1));
            }
        }
    }
}