士兵杀敌(二)
描述
南将军手下有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号士兵的总杀敌数,每组输出占一行
-
-
-
#include<stdio.h> int nu[10010]; typedef struct { int l,r; int sum; }node; node t[10010]; int q(int l,int r,int k) { if(t[k].l==l&&t[k].r==r) return t[k].sum; int mid=(t[k].l+t[k].r)/2; if(r<=mid) q(l,r,2*k); else if(l>=mid+1) q(l,r,2*k+1); else { return q(l,mid,2*k)+q(mid+1,r,2*k+1); } } void insert(int pos,int num,int k) { if(t[k].l==pos&&t[k].r==pos) { t[k].sum+=num; return ; } int mid=(t[k].l+t[k].r)/2; if(pos<=mid) insert(pos,num,2*k); else insert(pos,num,2*k+1); t[k].sum=t[2*k].sum+t[2*k+1].sum; } void create(int l,int r,int k) { t[k].l=l; t[k].r=r; if(l==r) { t[k].sum=nu[l]; return ; } int mid=(l+r)/2; create(l,mid,2*k); create(mid+1,r,2*k+1); t[k].sum=t[2*k].sum+t[2*k+1].sum; } int main() { int n,m,l,r; char a[10]; scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&nu[i]); } create(1,n,1); while(m--) { scanf("%s",&a); scanf("%d%d",&l,&r); if(a[0]=='Q') printf("%d\n",q(l,r,1)); else insert(l,r,1); } return 0; }