nyoj116——士兵杀敌(二)

来源:互联网 发布:中国学术期刊网数据库 编辑:程序博客网 时间:2024/05/16 14:28
描述

南将军手下有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 61 2 3 4 5QUERY 1 3ADD 1 2QUERY 1 3ADD 2 3QUERY 1 2QUERY 1 5
样例输出
68820


解决方法:很明显是树状数组中的改点求和。

#include <iostream>#include <cstdio>#include <cstring>#define maxn 1000005using namespace std;int n,m;int c[maxn];int lowbit(int x){    return x&(-x);}int Query(int x)//查询前x个数的和{    int sum=0;    while(x>0)    {        sum+=c[x];        x-=lowbit(x);    }    return sum;}void add(int x,int val){    while(x<=n)    {        c[x]+=val;        x+=lowbit(x);    }}int main(){    scanf("%d%d",&n,&m);    int val;    for(int i=1; i<=n; i++)    {        scanf("%d",&val);        add(i,val);    }    char str[10];    int s,e;    while(m--)    {        scanf("%s%d%d",str,&s,&e);        if(str[0]=='Q')        {            printf("%d\n",Query(e)-Query(s-1));        }        else            add(s,e);    }    return 0;}



0 0
原创粉丝点击