NYoj 116士兵杀敌(二)区间求和,单点更新
士兵杀敌(二)
时间限制: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
和上个代码求区间最大值改为区间和即可,这里注意的是节点数要开成4*叶子节点。(原因:刚好存满N个节点的线段树有趋近于2N个节点,而为了防溢出,要建出第N+1个叶子节点,那么根节点要再开一大棵子树出来,所以是4N)
其实就是等比数列求和,当第n层为N+1个节点时,总和就是4倍,N就是两倍。可参考:http://blog.csdn.net/u012313382/article/details/38488137#include <bits/stdc++.h> using namespace std; const int MAXNODE = (1e6)*4+10; const int MAX = 1e6+10; struct NODE { int value; int left,right; }node[MAXNODE]; int father[MAX];//叶子节点的编号 void BuildTree(int i,int left,int right) { node[i].left = left; node[i].right = right; node[i].value = 0; if(left == right) { father[left] = i; return ; } BuildTree(i<<1,left,(int)(floor(left+right)/2.0)); BuildTree((i<<1)+1,(int)(floor(left+right)/2.0+1),right); } //向上更新,更新所有区间和,传入一个加一个 void UpdateTree(int ri,int vl) { if(ri ==1 ) { node[ri].value += vl; return; } node[ri].value += vl; UpdateTree(ri/2,vl); // int fi = ri/2; // int a = node[fi<<1].value; // int b = node[(fi<<1)+1].value; // node[fi].value = a + b; // UpdateTree(ri/2); } int sum; //查询从上向下查找 void Query(int i,int l,int r) { if(node[i].left == l&&node[i].right == r) { sum += node[i].value; return; } //儿子左子树 i = i<<1; //分段分割求 if(l <= node[i].right) { if(r <= node[i].right) Query(i,l,r); else Query(i,l,node[i].right); } //右子树 i++; if(r >= node[i].left) { if(l >= node[i].left) Query(i,l,r); else Query(i,node[i].left,r); } } int main() { // freopen("in.txt","r",stdin); int n,m,g; ios::sync_with_stdio(false); cin>>n>>m; //建空树 BuildTree(1,1,n); for(int i = 1; i <= n; i++) { cin>>g; UpdateTree(father[i],g); } string op; int a,b; while(m--) { cin>>op>>a>>b; if(op[0] == 'Q') { sum = 0; Query(1,a,b); cout<<sum<<endl; } else { //传入节点和增加的值 UpdateTree(father[a],b); } } return 0; }