敌兵布阵-线段树（1）

A - 敌兵布阵

Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Submit Status

Description

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营 地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工 兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek 问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而 Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死 肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算 机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我 知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果 你的程序效率不够高的话，Tidy还是会受到Derek的责骂的.

 

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

Sample Output

Case 1:

6

33

59

 

是线段树的模板题

把区间设为自己到自己！

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100010<<2//一棵树的某排有N个数，那此排以前就一共有N-1个数，且此排的下一排有2N个数，N+N-1+2N约等于4N，要开N的四倍数组，才不会越界
int sto[N],up[N],x,y;
char s[3]; 8 void init(int t,int w,int o)//初始化树中的某一结点，t为该结点左边界，w为右边界，o为该结点的标号（用来表示该结点的位置）
{
    up[o]=0;//up[o]代表该结点的值需要改变时，增加或减少的量；
    if(t==w)//当递归到最底层，即划分到区间长度为0时
    {
        scanf("%d",&sto[o]);//输出该结点的数值
        return;
    }
    int mid=(t+w)>>1;//除以2，划分下一层的边界
    int lso=o<<1;//乘以二，求出其左子数的标号
    init(t,mid,lso);//递归到其左子树
    init(mid+1,w,lso|1);
    sto[o]=sto[lso]+sto[lso|1];//该结点的左子树的值加上右子树的值，从下到上进行累加，获得整棵数的值得累加和
}
   //某个结点代表的是某个范围，此结点想要变化N，就代表这一范围的每个数都要变化N；
void updown(int sum,int o)//更新某个结点的值的操作，sum指该结点以下有多少个结点（他们同样要一起更新），o代表该结点的标号
{//第一次修改值得时候，是不会用到的，因为会有多组测试数据，会进行多次修改，所以才要更新
    if(up[o])//如果变化量不为0，则需要进行更新
    {
        int lso=o<<1;//此结点的左子树标号
        up[lso]+=up[o];//把变化量传给左子数
        up[lso|1]+=up[o];//把变化量传给右子数
        sto[lso]+=(sum-(sum>>1))*up[o];//左子树的值为它自身加上它以下的所有结点数变化量之和（左子树的值要多1，所以要加多的那一边）
        sto[lso|1]+=(sum>>1)*up[o];
        up[o]=0;//用完了就制为0，以免后面重复加
    }
}
int updata(int t,int w,int o,int st,int en,int num)//查找并计算出某颗树的值，t,w为整棵树的范围，o,代表查找的位置，st，en，代表该位置的范围，num代表变化量
{
    if(st<=t&&w<=en)//当查到那个那个位置的范围以内时
    {
        if(num!=1<<29)//如果num不为无穷大，即除了查询操作，还要进行更新工作
        {
            up[o]+=num;//把要加的变量付给up[o]
            sto[o]+=(w-t+1)*num;//算出该范围总的变化量
        }
    return sto[o];//返回求和
    }
    updown(w-t+1,o);//继续向下更新
    int mid=(t+w)>>1;//继续缩小范围
    int lso=o<<1;
    int ans=0;
    if(st<=mid) ans+=updata(t,mid,lso,st,en,num);//当st小于mid 则表明该范围至少有一部分在左子树；则减小范围，再次进行查找计算，然后一层一层退回；
    if(en>mid) ans+=updata(mid+1,w,lso|1,st,en,num);//当en>mid 说明，该范围至少有一部分在右子树，则减小范围，再次进行查找计算，然后层层退回；
    sto[o]=sto[lso]+sto[lso|1];//求出每一个范围的总值;
    return ans;//每次返回回去都不再重新走ans=0，ans一直累加，直到输出答案
}
int main()
{
    int n,m,t=1;
    scanf("%d",&n);
    while(n--)
    {
        scanf("%d",&m);
        init(1,m,1);
        printf("Case %d:\n",t++);
        while(1)
        {
            scanf("%s",s);
            if(s[0]=='Q')
            {
                scanf("%d%d",&x,&y);
                printf("%d\n",updata(1,m,1,x,y,1<<29));//把查询范围x到y的和为多少
            }
            if(s[0]=='A')
            {
                scanf("%d%d",&x,&y);
                updata(1,m,1,x,x,y);//把范围订在自己到自己
            }
            if(s[0]=='S')
            {
                scanf("%d%d",&x,&y);
                updata(1,m,1,x,x,-y);
            }
            if(s[0]=='E')
                break;
        }
    }
    return 0;
}

 

 

转载于:https://www.cnblogs.com/tianmin123/p/4757953.html