士兵杀敌(二)
时间限制:
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<cstdio>
int a[1000010];
struct node
{
int left,right,sum;
}num[5000000];
int build(int left,int right,int cnt)
{
int mid;
num[cnt].left=left;
num[cnt].right=right;
num[cnt].sum=0;
mid=(left+right)>>1;
if(left==right)
return num[cnt].sum=a[left];
else
return num[cnt].sum=build(left,mid,cnt*2)+build(mid+1,right,cnt*2+1);
}
int query(int left,int right,int cnt)
{
int mid;
if(num[cnt].left==left&&num[cnt].right==right)
return num[cnt].sum;
mid=(num[cnt].left+num[cnt].right)>>1;
if(right<=mid)
return query(left,right,cnt*2);
else if(left>mid)
return query(left,right,cnt*2+1);
else
return query(left,mid,cnt*2)+query(mid+1,right,cnt*2+1);
}
int add(int pos,int count,int cnt)
{
int mid;
num[cnt].sum+=count;
if(num[cnt].left==num[cnt].right)
return 0;
mid=(num[cnt].left+num[cnt].right)>>1;
if(pos<=mid)//注意是小于且等于
add(pos,count,cnt*2);
else
add(pos,count,cnt*2+1);
}
int main()
{
int n,m,i,b,d;
char s[10];
while(scanf("%d%d",&n,&m)!=EOF)
{
for(i=1;i<=n;++i)
scanf("%d",&a[i]);
build(1,n,1);
while(m--)
{
scanf("%s%d%d",s,&b,&d);
if(s[0]=='Q')
printf("%d\n",query(b,d,1));
else
add(b,d,1);
}
}
return 0;
}