描述
南将军手下有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> typedef struct { int l,r; int sum; }node; node t[4*1000000]; int nu[1000005]; void creat(int l,int r,int k); int q(int l,int r,int k); void insert(int pos,int num,int k); int main() { int N,M; int i,num,l,r; char qu[8]; scanf("%d%d",&N,&M); for(i=1;i<=N;++i) { scanf("%d",&nu[i]); } creat(1,N,1); while(M--) { scanf("%s",qu); scanf("%d%d",&l,&r); if(qu[0]=='Q') printf("%d\n",q(l,r,1)); else insert(l,r,1); } } void creat(int l,int r,int k) { t[k].l=l; t[k].r=r; // t[k].sum=0; if(t[k].l==t[k].r) { t[k].sum=nu[l]; return; } int mid=(t[k].l+t[k].r)/2; creat(l,mid,2*k); creat(mid+1,r,2*k+1); t[k].sum=t[2*k].sum+t[2*k+1].sum; } 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) return q(l,r,2*k); else if(l>=mid+1) return 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; }