poj 1061 士兵杀敌

树状数组。

模板

int lowbit(int x)
{
    return x & (-x);
}
void modify(int x,int add)//一维
{  
    while(x<=MAXN)  
    {      
        a[x]+=add;    
        x+=lowbit(x); 
    }
}
int get_sum(int x)
{  
    int ret=0; 
    while(x!=0)  
    {       
        ret+=a[x];   
        x-=lowbit(x);   
    }  
    return ret;
}

这样把查询，增加的复杂度变为logn  

但是如果用c++的流输入输出。800+ms   快超时了，

换成c的 就300+ms  

还是用c 吧以后。。。 好危险。

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 50020
#define LL long long
using namespace std;
int n,c[N];
int lowbit(int x)
{
    return x&(-x);
}
void add(int i,int val)
{
    while(i<=n)
    {
        c[i]+=val;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=c[i];
        i-=lowbit(i);
    }
    return s;
}
int main()
{
    int T,t,ca=1;
    char str[10];
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        printf("Case %d:\n",ca++);
        memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&t);
            add(i,t);
        }

        while(1)
        {
            int x,y;
            scanf("%s",str);
            if(str[0]=='E') break;
            if(str[0]=='Q')
            {
                scanf("%d%d",&x,&y);
                printf("%d\n",sum(y)-sum(x-1));
            }
            else if(str[0]=='A')
            {
                scanf("%d%d",&x,&y);
                add(x,y);
            }
            else if(str[0]=='S')
            {
                scanf("%d%d",&x,&y);
                add(x,-y);
            }
            //else break;
        }



    }
    return 0;
}