HDU 1166 敌兵布阵 （线段树：单点增减求和）

敌兵布阵

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 78396    Accepted Submission(s): 33118

Problem Description

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​​​ ​​1754​​​ ​​1542​​​ ​​1540​​ 

题解：区间查询和修改。典型的线段树。

AC代码：

#pragma comment(linker, "/STACK:102400000,102400000")
//#include<bits/stdc++.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <map>
#include <cmath>
#include <queue>
#include <set>
#include <bitset>
#include <iomanip>
#include <list>
#include <stack>
#include <utility> 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef vector<int> vi;
const double eps = 1e-8;  
const int INF = 1e9+7; 
const ll inf =(1LL<<62) ;
const int MOD = 1e9 + 7;  
const ll mod = (1LL<<32);
const int N =1e6+6; 
const int M=100010;
const int maxn=55555;
#define mst(a) memset(a, 0, sizeof(a))
#define M_P(x,y) make_pair(x,y)
#define in freopen("in.txt","r",stdin) 
#define rep(i,j,k) for (int i = j; i <= k; i++)  
#define per(i,j,k) for (int i = j; i >= k; i--)  
#define lson  l , mid , rt << 1    
#define rson  mid + 1 , r , rt << 1 | 1  
const int lowbit(int x) { return x&-x; }
//const int lowbit(int x) { return ((x)&((x)^((x)-1))); } 
int read(){ int v = 0, f = 1;char c =getchar();
while( c < 48 || 57 < c ){if(c=='-') f = -1;c = getchar();}
while(48 <= c && c <= 57) v = v*10+c-48, c = getchar();
return v*f;}
int sum[maxn<<2];

void pushup(int rt) //把当前结点的信息更新到父结点
{
  //线段树是用数组来模拟树形结构
  //对于每一个节点rt,左子节点为 2*rt (一般写作rt<<1)右子节点为 2*rt+1（一般写作rt<<1|1） 
  sum[rt] = sum[rt<<1] + sum[rt<<1|1]; 
}
void build(int l,int r,int rt)
{
  if(l==r)
  {
    scanf("%d",&sum[rt]); return;
  }
  int mid=(l+r)>>1;
  build(lson);//递归构造左子树
  build(rson);//递归构造右子树
  pushup(rt); //更新求和 
}
void update(int p, int add,int l,int r,int rt)//单点增减 
{
  if(l==r){
    sum[rt]+=add; return ;
  }
  int mid=(l+r)>>1;
  if(p<=mid) update(p,add,lson);
  else update(p,add,rson);
  pushup(rt);
}
int query(int L,int R,int l,int r,int rt) //区间求和 
{
  if(L <= l && r <= R){
    return sum[rt];
  }
  int mid = (l+r)>>1;
  int ans = 0;
  if(L <= mid) ans+=query(L,R,lson);
  if(R > mid) ans+=query(L,R,rson);
  return ans; 
}
int main()
{
  int T,n;
  int cas=1;
  scanf("%d",&T);
  while(T--)
  {
    printf("Case %d:\n",cas++);  
    cin>>n;
    build(1,n,1);
    char op[10];
    while(scanf("%s",op)) {
      if(op[0]=='E')break;
      int a,b;
      scanf("%d%d",&a,&b);
      if(op[0]=='Q')printf("%d\n",query(a,b,1,n,1));
      else if(op[0]=='S')update(a,-b,1,n,1);
      else update(a,b,1,n,1);
    }
  }
      return 0;
}