HDU 1166 敌兵布阵 （线段树单点修改和区间和查询）

Input

第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1)
Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j
,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j
,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End
表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令

 

Output

对第i组数据,首先输出“Case
i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

 

Sample Input

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End

 

#include<cstdio>
#include<algorithm>
#include<cstring>
#define ls x<<1
#define rs x<<1|1
using namespace std;
const int N=;
int tr[N<<];
int T,Case,n;

void bt(int l,int r,int x)
{
    if (l==r)
    {
        int dat;
        scanf ("%d",&dat);
        tr[x]=dat; return;
    }
    int mid=(l+r)>>;
    bt(l,mid,ls);
    bt(mid+,r,rs);
    tr[x]=tr[ls]+tr[rs];
    return;
}

void add(int q,int a,int l,int r,int x)
{
    if (l==r)
    {
        tr[x]+=a;
        return;
    }
    int mid=(l+r)>>;
    if (q<=mid) add(q,a,l,mid,ls);
    else add(q,a,mid+,r,rs);
    tr[x]=tr[ls]+tr[rs];
    return;
}

void sub(int q,int a,int l,int r,int x)
{
    if (l==r)
    {
        tr[x]-=a;
        return ;
    }
    int mid=(l+r)>>;
    if (q<=mid) sub(q,a,l,mid,ls);
    else sub(q,a,mid+,r,rs);
    tr[x]=tr[ls]+tr[rs];
    return ;
}

int que(int ql,int qr,int l,int r,int x)
{
    if (ql<=l && qr>=r)
    {
        return tr[x];
    }
    int mid=(l+r)>>,ans=;
    if (ql<=mid) ans+=que(ql,qr,l,mid,ls);
    if (qr>mid) ans+=que(ql,qr,mid+,r,rs);
    return ans;
}

int main()
{
    scanf ("%d",&T);
    while (T--)
    {
        Case++;
        printf("Case %d:\n",Case);
        memset(tr,,sizeof(tr));
        scanf ("%d",&n);
        bt(,n,);
        char s[];
        while ()
        {
            getchar();
            scanf ("%s",s);
            if (s[]=='Q')
            {
                int l,r;
                scanf ("%d%d",&l,&r);
                printf("%d\n",que(l,r,,n,));
            }
            else if (s[]=='A')
            {
                int x,y;
                scanf ("%d%d",&x,&y);
                add(x,y,,n,);
            }
            else if (s[]=='S')
            {
                int x,y;
                scanf ("%d%d",&x,&y);
                sub(x,y,,n,);
            }
            else break;
        }
    }
    return ;
}

Sample Output

Case 1:
6
33
59

 

 

十分基础的线段树模板，存一下。