CCNU-线段树练习题-A-单点更新1

A - 单点更新1

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<iostream>
#include<string>
using namespace std;
struct BTs
{
    int e;
    int r;
    int l;
};
const int M = 50002*10;
BTs a[M];
int b[50010];


inline void init(int k,int h,int x)   //k 为左边界 h为右边界  x 数组下标
{
    a[x].r = h;                       //从叶子节点自下而上建树，叶子节点即是最小不可分的区间
    a[x].l = k;
    if(k == h)
    {
       a[x].e = b[k];    
       return ;
    }
    else
    {
        init(k,(k+h)/2,2*x);
        init((k+h)/2+1,h,2*x+1);
        a[x].e = a[2*x].e + a[2*x+1].e;
    }
}
inline void change_plus(int index,int addend,int k)     //k数组下标
{
    if(a[k].l <= index&&a[k].r >=index)                //更新了所有覆盖的区域
    {                                                  
        a[k].e += addend;
        change_plus(index,addend,k*2);
        change_plus(index,addend,k*2+1);
    }
    else return ;
}
inline void change_sub(int index,int m,int k)
{
    if(a[k].l <= index&&a[k].r >=index)
    {
        a[k].e -= m;
        change_sub(index,m,k*2);
        change_sub(index,m,k*2+1);
    }
    else return ;
}
int count;
inline void query(int L,int R,int k)
{
    if(L == a[k].l&&R == a[k].r)                             //找到了相应的区间，计数；

    {
        count += a[k].e;
       // cout<<"\""<<k<<"\""<<count<<endl;
        return;
    }
    else if(L <= (a[k].l + a[k].r)/2&&R >= (a[k].r + a[k].l)/2+1)     //在搜索区间值的时候把区间边界和对应区间边界的数组坐标弄混了
    {
        query(L,(a[k].l+a[k].r)/2,k*2);
        query((a[k].l+a[k].r)/2+1,R,k*2+1);
    }
    else if(R <= (a[k].r+a[k].l)/2) query(L,R,k*2);
    else if(L >= (a[k].l+a[k].r)/2+1) query(L,R,k*2+1);
}
int main()
{
    int T;
    cin>>T;
    int rc = T;
    while(T)
    {
        cout<<"Case "<<rc-T+1<<":"<<endl;
        int c;
        cin>>c;


        for(int i = 1; i <= c; i++)
          cin>>b[i];

        init(1,c,1);
//for(int i = 1; i < 4*c; i++)
  //cout<<"\""<<i<<"\""<<a[i].e<<endl;
        while(1)
        {
            string s;
            int i,j;
            cin>>s;

              if(s == "Sub")
              {
                  cin>>i>>j;
                  change_sub(i,j,1);
              }
              if(s == "Add")
              {
                  cin>>i>>j;
                  change_plus(i,j,1);
              }
              if(s == "Query")
              {
                  cin>>i>>j;
                  count = 0;
                  query(i,j,1);
                  //cout<<query(i,j,1)<<endl;
                  cout<<count<<endl;
              }
              if(s == "End")
              {
                  break;
              }

        }
      T--;
    }
    return 0;
}
           

这一题我没有套模板去做，只是了解了线段树的概念之后，全部自己琢磨著敲的。主要是想看看自己的思路哪里有差距，也正好让自己多犯点错。 但很明显，这种做法的效率不高。空间占用也大，空间的最坏情况是O（4n），但不知道为什么我开到5n还是数组越界，然后一怒开到了20n~~o(╯□╰)o 等我把模板好好看看再更新T^T【ps.花咗我兩日時間，真系傷心個啫】~