【线段树I：母题】hdu 1166 敌兵布阵

时间：2022-11-05 05:12:00

【线段树I：母题】hdu 1166 敌兵布阵

题目链接：hdu 1166 敌兵布阵

题目大意

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的责骂的.

笔者生平第一道线段树。线段树是一种高级的二叉搜索树，建树、更新、查询操作复杂度均为O(logN)

【线段树I：母题】hdu 1166 敌兵布阵

说一下思路

  • 线段树的点改动、区间查询
  • segTree[4*_max]四倍数据数组大小

參考代码

/*Author:Hacker_vision*/

#include<bits/stdc++.h>

using namespace std;

const int _max = 50000 + 10;

int n,a[_max],x,y;

char s[20];

struct segTree{

  int lc,rc;//左孩子、右孩子

  int sum;

}segTree[_max*4];//线段树一般为4倍数组元素个数

void build(int root,int L,int R){//建树 O(logN)

   segTree[root].lc=L;

   segTree[root].rc=R;

   if(L==R){

    segTree[root].sum=a[L];

    return;

   }

   int mid = (L + R) >> 1;

   build(root<<1,L,mid);

   build(root<<1|1,mid+1,R);

   segTree[root].sum=segTree[root<<1].sum+segTree[root<<1|1].sum;//更新父节点

}

void update(int root,int pos,int add){//点改动O(logN)

  if(segTree[root].lc==segTree[root].rc){

    segTree[root].sum += add;

    return ;

  }

  int mid = (segTree[root].lc + segTree[root].rc) >> 1;

  if(pos <= mid) update(root<<1,pos,add);

  else update(root<<1|1,pos,add);

  //回溯更新父节点

  segTree[root].sum=segTree[root<<1].sum+segTree[root<<1|1].sum;

}

int query(int root,int L,int R){//区间查询，O(logN)

  if(segTree[root].lc==L && segTree[root].rc == R) return segTree[root].sum;

  int mid = (segTree[root].lc + segTree[root].rc)>>1;

  if(R <= mid) return query(root<<1,L,R);

  else if(mid < L) return query(root<<1|1,L,R);

  else return query(root<<1,L,mid)+query(root<<1|1,mid+1,R);

}

int main(){

 //freopen("input.txt","r",stdin);

 int T;cin>>T;int cnt=1;

 while(T--){

   scanf("%d",&n);

   for( int i = 1; i <= n; ++ i) scanf("%d",a+i);

   build(1,1,n);//建立线段树

   printf("Case %d:\n",cnt++);

   while(scanf("%s",s)==1&&strcmp(s,"End")){

     scanf("%d%d",&x,&y);

     if(strcmp(s,"Add")==0) update(1,x,y);//点改动

     else if(strcmp(s,"Sub")==0) update(1,x,-y);

     else if(strcmp(s,"Query")==0) printf("%d\n",query(1,x,y));//查询区间Ax+……+Ay

   }

 }

 return 0;

}
  • 加粗 Ctrl + B
  • 斜体 Ctrl + I
  • 引用 Ctrl + Q
  • 插入链接 Ctrl + L
  • 插入代码 Ctrl + K
  • 插入图片 Ctrl + G
  • 提升标题 Ctrl + H
  • 有序列表 Ctrl + O
  • 无序列表 Ctrl + U
  • 横线 Ctrl + R
  • 撤销 Ctrl + Z
  • 重做 Ctrl + Y

相关文章