HDU 1166 敌兵布阵 ——————线段树，区间求和

柔光的暖阳◎ 2022-05-14 23:57 226阅读 0赞

# 敌兵布阵 #

**Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)  
Total Submission(s): 123548 Accepted Submission(s): 51764**

Problem 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

Author

Windbreaker

Recommend

Eddy | We have carefully selected several similar problems for you: 1394 1698 1542 1540 1255

Statistic | Submit | Discuss | Note

--------------------

#include<bits/stdc++.h>
    using namespace std;
    const int MAXNODE=1<<20;
    const int MAXN=2e6+3;
    struct NODE{
        int l,r,v;
    }node[MAXNODE];
    int father[MAXN];
    
    void bt(int i,int l,int r)//建树0
    {
        node[i].l=l;
        node[i].r=r;
        node[i].v=0;
        if(l == r)
        {
            father[l]=i;
            return ;
        }
        bt(i<<1, l, (l+r)/2);
        bt(i<<1|1, (l+r)/2 + 1, r);
    }
    void update(int ri)//更新树
    {
        if(ri == 1) return ;
        int fi = ri/2;
        int a=node[fi<<1].v;
        int b=node[fi<<1|1].v;
        node[fi].v= a+b ;
        update(ri/2);
    }
    int Max;
    void query(int i,int l,int r)//要查寻的区间[l,r];
    {
        if(node[i].l==l && node[i].r == r)// 第i个结点 刚好保存的是[l,r]的最大值
        {
            Max +=  node[i].v;
            return ;
        }
        i<<=1;//左儿子结点
        if(l <= node[i].r)
        {
            if(r <= node[i].r)  query(i,l,r);
            else        query(i,l,node[i].r);
        }
        i++;//右儿子结点
        if(r >= node[i].l)
        {
            if(l >= node[i].l)  query(i,l,r);
            else        query(i,node[i].l,r);
        }
    }
    
    
    int main()
    {
        int t,n,m,g;
        ios::sync_with_stdio(false);
        cin>>t;
        int num=1;
        while(t--)
        {
            cin>>n;
            bt(1,1,n);
            for(int i=1;i<=n;i++)
            {
                cin>>g;
                node[father[i] ].v = g;
                update(father[i]);
            }
            string op;
            int i,j;
            cout<<"Case "<<num++<<":"<<endl;
            while(cin>>op && op[0]!='E')
            {
                cin>>i>>j;
                if(op[0]=='Q')
                {
                    Max=0;
                    query(1,i,j);
                    cout<<Max<<endl;
                }
                if(op[0]=='A')
                {
                    node[father[i] ].v += j;
                    update(father[i]);
                }
                if(op[0]=='S')
                {
                    node[father[i] ].v -= j;
                    update(father[i]);
                }
            }
    
        }
        while(cin>>n>>m)
        {
            bt(1,1,n);
            for(int i=1; i<=n; i++)
            {
                cin>>g;
                node[father[i] ].v = g;
                update(father[i]);
            }
            string op;
            int a,b;
            while(m--)
            {
                cin>>op>>a>>b;
                if(op[0] == 'Q')
                {
                    Max=0;
                    query(1,a,b);
                    cout<<Max<<endl;
                }
                else
                {
                    node[father[a] ].v = b;
                    update(father[a]);
                }
            }
    
        }
        return 0;
    }

白书代码修改的

/**
     *　　　　　　　　┏┓　　　┏┓+ +
     *　　　　　　　┏┛┻━━━┛┻┓ + +
     *　　　　　　　┃　　　　　　　┃
     *　　　　　　　┃　　　━　　　┃ ++ + + +
     *　　　　　　 ████━████ ┃+
     *　　　　　　　┃　　　　　　　┃ +
     *　　　　　　　┃　　　┻　　　┃
     *　　　　　　　┃　　　　　　　┃ + +
     *　　　　　　　┗━┓　　　┏━┛
     *　　　　　　　　　┃　　　┃
     *　　　　　　　　　┃　　　┃ + + + +
     *　　　　　　　　　┃　　　┃　　　　Code is far away from bug with the animal protecting
     *　　　　　　　　　┃　　　┃ + 　　　　神兽保佑,代码无bug
     *　　　　　　　　　┃　　　┃
     *　　　　　　　　　┃　　　┃　　+
     *　　　　　　　　　┃　 　　┗━━━┓ + +
     *　　　　　　　　　┃ 　　　　　　　┣┓
     *　　　　　　　　　┃ 　　　　　　　┏┛
     *　　　　　　　　　┗┓┓┏━┳┓┏┛ + + + +
     *　　　　　　　　　　┃┫┫　┃┫┫
     *　　　　　　　　　　┗┻┛　┗┻┛+ + + +
     */
    /*--------- Hongjie ----------*/
    // #include<bits/stdc++.h>
    #include<map>
    #include<set>
    #include<queue>
    #include<stack>
    #include<deque>
    #include<cmath>
    #include<cstdio>
    #include<bitset>
    #include<string>
    #include<vector>
    #include<cstring>
    #include<iostream>
    #include<algorithm>
    using namespace std;
    
    typedef long long ll;
    typedef pair<int ,int> P;
    
    const int INF  = 0x3f3f3f3f;
    const int MAXN = 3e5+7;
    int dat[MAXN];
    int n,m,t;
    
    void init(int _n) {
        
            n = 1;
            while(n<_n)     n<<=1;
            memset(dat,0,sizeof(int)*(n*4));
    }
    
    void update(int k,int a) {
        
            k += n-1;
            dat[k] += a;
            while(k>0) {
        
                    k = (k-1)/2;
                    dat[k] = dat[k*2+1]+dat[k*2+2];
            }
    }
    int query(int a,int b,int k,int l,int r) {
        
            if(r<=a || b<=l)        return 0;
            if(a<=l && b>=r)        return dat[k];
            else {
        
                    int vl = query(a,b,k*2+1,l,(l+r)/2);
                    int vr = query(a,b,k*2+2,(l+r)/2,r);
                    return vl+vr;
            }
    }
    
    
    int main(){
        
            //freopen("../in.txt","r",stdin);
            // freopen("../out.txt","w",stdout);
            ios::sync_with_stdio(0);
            cin.tie(0);
            cin>>t;
            int Case = 0;
            while(t--){
        
                    cin>>n;
                    int n_ = n;
                    init(n);
                    for(int i=0;i<n_;++i){
        
                            int x;
                            cin>>x;
                            update(i,x);
                    }
    
                    string s;
                    int i,j;
                    cout<<"Case "<<++Case<<":"<<endl;
                    while(true) {
        
                            cin>>s;
                            if(s[0]=='A')   cin>>i>>j,update(i-1,j);
                            if(s[0]=='S')   cin>>i>>j,update(i-1,-j);
                            if(s[0]=='Q')   cin>>i>>j,cout<<query(i-1,j,0,0,n)<<endl;
                            if(s[0]=='E')   break;
                    }
            }
            return 0;
    }