[原]kuangbin——线段树专题 A - 敌兵布阵

楚东方 16/08/04 11:45:38

A - 敌兵布阵
Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Description

Lily 特别喜欢养花,但是由于她的花特别多,所以照料这些花就变得不太容易。她把她的花依次排成一行,每盆花都有一个美观值。如果Lily把某盆花照料的好的话,这盆花的美观值就会上升,如果照料的不好的话,这盆花的美观值就会下降。有时,Lily想知道某段连续的花的美观值之和是多少,但是,Lily的算术不是很好,你能快速地告诉她结果吗?

Input

	第一行一个整数T,表示有T组测试数据。
每组测试数据的第一行为一个正整数N(N<=50000),表示Lily有N盆花。接下来有N个正整数,第i个正整数ai表示第i盆花的初始美观值(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 i j"命令,输出第i盆花到第j盆花的美观值之和。

Sample Input

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

Sample Output

Case 1:
133050

/*************************************************************************
	> File Name: A.cpp
	> Author:chudongfang 
	> Mail:1149669942@qq.com 
	> Created Time: 2016年08月04日 星期四 08时17分23秒
 ************************************************************************/

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include <algorithm>
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;

#define M 50005
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1


int sum[M<<2];

void pushplus(int rt)
{
    sum[rt] = sum[rt<<1] + sum [rt<<1|1];
}

void build(int l,int r,int rt)
{
    if(l == r){
        scanf("%d",&sum[rt]);
        return;
    }

    int m = (l+r)>>1;
    build(lson);
    build(rson);
    pushplus(rt);
    
}


void update(int p,int add,int l,int r ,int rt)
{
    if(l == r)
    {
        sum[rt] += add;
        return ;
    }
    int m = (l+r)>>1;
    
    if(p<=m)
        update(p,add,lson);
    else
        update(p,add,rson);

    pushplus(rt);
}


int query(int L,int R,int l,int r ,int rt)
{
    if(L<=l && r<=R)
    {
        return sum[rt];
    }
    int m = (l + r) >>1;
    int ans = 0;
    if(L<=m)
        ans += query(L,R,lson);
    if(R>m)
        ans += query(L,R,rson);
    
    return ans;
}




int main(int argc,char *argv[])
{
    int T,a,b,n;
    scanf("%d",&T);
    for(int i=1;i<=T;i++)
    {
        printf("Case %d:\n",i);
        scanf("%d",&n);
        build(1,n,1);
        char op[100];
        while(1)
        {
            scanf("%s",op);
            if(strcmp(op,"End") == 0)
                break;
            if(strcmp(op,"Query") == 0)
            {
                scanf("%d %d",&a,&b);
                printf("%d\n",query(a,b,1,n,1));
            }
            else if(strcmp(op,"Add") == 0)
            {
                scanf("%d %d",&a,&b);
                update(a,b,1,n,1);
            }
            else if(strcmp(op ,"Sub") == 0)
            {
                scanf("%d %d",&a,&b);
                update(a,-b,1,n,1);
            }
        }
    }
    return 0;
}





作者:chudongfang2015 发表于2016/8/4 11:45:38 原文链接
阅读:7 评论:0 查看评论