线段树FB

hdu 1166 敌兵布阵

最简单的单点更新，因为是第一题线段树，学习了网上的题解之后写的

#include<cstdio>
#include<iostream>
using namespace std;

const int maxn = 50000;

struct segment{
     int l,r,sum;
} T[maxn*4];

void build(int l,int r,int k)
{
    T[k].l = l;
    T[k].r = r;
    T[k].sum = 0;
    if(l == r) return;
    int mid = (l + r) >> 1;
    build(l, mid, 2*k);
    build(mid+1, r, 2*k+1);
}

//查询
int ans;
void query(int k, int l, int r)
{
    if(T[k].l == l && T[k].r == r)
    {
        ans += T[k].sum;
        return ;
    }
    int mid = (T[k].l + T[k].r) >> 1;
    if(r<=mid)       query(2*k,l,r);
    else if(l>mid)   query(2*k+1,l,r);
    else
    {
        query(2*k,l,mid);
        query(2*k+1,mid+1,r);
    }
    return ;
}

void update(int k, int d,int n)
{
    if(T[k].l == T[k].r && T[k].l == d)
    {
        T[k].sum += n;      // 叶节点， 更新sum
        return ;
    }
    int mid = (T[k].l + T[k].r) >> 1;
    if(d <= mid) update(k*2, d, n); else update(k*2+1, d, n);
       //更新本结点的sum
    T[k].sum = T[k*2].sum + T[k*2+1].sum;
}



int main()
{
    int t, Case = 0;
    scanf("%d",&t);
    while(t--)
    {
        int n,x;
        scanf("%d",&n);
        build(1,n,1);
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&x);
            update(1,i,x);
        }
        printf("Case %d:
",++Case);
        char str[11];
        while(~scanf("%s",str) && str[0] != 'E')
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(str[0] == 'Q')
            {
                 ans = 0;
                 query(1,a,b);
                 printf("%d
",ans);
            }
            if(str[0] == 'A')
                 update(1,a,b);
            if(str[0] == 'S')
                 update(1,a,-b);
        }
    }
    return 0;
}