码迷,mamicode.com
首页 > 其他好文 > 详细

敌兵布阵-线段树(1)

时间:2015-08-25 18:36:40      阅读:237      评论:0      收藏:0      [点我收藏+]

标签:

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;
}

 

 

敌兵布阵-线段树(1)

标签:

原文地址:http://www.cnblogs.com/tianmin123/p/4757953.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!