HDU1166敌兵布阵-线段树求和

HDU1166敌兵布阵-线段树求和

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
using namespace std;
const int maxn = 200005*2;
long long n,dat[2*maxn-1];
void init(int n_)
{
    n = 1;
    while(n<n_)n*=2;
    for(int i = 0 ;i < 2*n-1;i++)
        dat[i]= 0;
}
void update(int k,int a)
{
    k += n-1;
    dat[k] += a;
    while(k>0)
    {
        k = (k-1)/2;
        dat[k]+=a;
    }
}
int query(int a, int b,int k,int l ,int r)
//查询用 query(a,b,0,0,n)·查询[a,b)
{
    if(r<=a||b<=l){return 0;}
    if(a<=l&&r<=b)return dat[k];
    else
    {
        int vl = query(a,b,k*2+1,l,(l+r)/2);
        int vr = query(a,b,k*2+2,(l+r)/2,r);
        return vl+vr;
    }
}
int main()
{
    int m,a;
    char op[10];
    cin>>m;
    for(int k = 1; k <= m ; k++)
    {
        cout<<"Case "<<k<<":"<<endl;
        scanf("%d",&n);
        init(n);
        for(int i = 0 ; i < n; i++)
        {
            scanf("%d",&a);
            update(i,a);
        }
        int op1,op2;
        scanf("%s",op);
        while(strcmp(op,"End"))
        {
            scanf("%d %d",&op1,&op2);
            if(strcmp(op,"Query")==0)
            {
                printf("%d\n",query(op1-1,op2,0,0,n));
            }
            else if(strcmp(op,"Add")==0)
            {
                update(op1-1,op2);
            }
            else if(strcmp(op,"Sub")==0)
            {
                update(op1-1,-op2);
            }
            getchar();
            scanf("%s",op);
        }
    }
    return 0;
}