（线段树）hdoj1166-敌兵布阵

69 阅读 0 评论 46 点赞

我是靠谱客的博主悦耳鱼，最近开发中收集的这篇文章主要介绍（线段树）hdoj1166-敌兵布阵，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

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

线段树的基础应用，init函数初始化树各个结点，add，sub函数分别处理对线段树上某叶子结点的值的更新，query函数返回查询区间的值。
参考代码

 1 #include <iostream>
 2 #include <queue>
 3 #include <cstdio>
 4 #include <cstring>
 5 #include <algorithm>
 6 using namespace std;
 7 typedef long long ll;
 8 typedef unsigned long long ull;
 9 struct node
10 {
11
int left,right,mid;
12
int num;
13 }p[500005];
14 void init(int a,int b,int n)
15 {
16
p[n].left=a;
17
p[n].right=b;
18
p[n].mid=(a+b)/2;
19
p[n].num=0;
20
if(a+1==b)
21
return;
22
init(a,(a+b)/2,2*n);
23
init((a+b)/2,b,2*n+1);
24 }
25 void add(int pos,int value,int n)
26 {
27
p[n].num+=value;
28
if(p[n].left+1==p[n].right)
29
return;
30
if(pos<p[n].mid)
31
add(pos,value,2*n);
32
else
33
add(pos,value,2*n+1);
34 }
35 void sub(int pos,int value,int n)
36 {
37
p[n].num-=value;
38
if(p[n].left+1==p[n].right)
39
return;
40
if(pos<p[n].mid)
41
sub(pos,value,2*n);
42
else
43
sub(pos,value,2*n+1);
44 }
45 int query(int a,int b,int n)
46 {
47
if(p[n].left==a&&p[n].right==b)
48
return p[n].num;
49
if(a<p[n].mid)
50 
{
51
if(b<=p[n].mid)
52
return query(a,b,2*n);
53
else
54
return query(a,p[n].mid,2*n)+query(p[n].mid,b,2*n+1);
55 
}
56
else
57
return query(a,b,2*n+1);
58 }
59 int t;
60 int N;
61 int s;
62 char rec[10];
63 int main()
64 {
65
scanf("%d",&t);
66
for(s=1;s<=t;s++)
67 
{
68
printf("Case %d:n",s);
69
scanf("%d",&N);
70
init(1,N+1,1);
71
int i,tem,j;
72
for(i=1;i<=N;i++)
73 
{
74
scanf("%d",&tem);
75
add(i,tem,1);
76 
}
77
while(scanf("%s",rec)&&rec[0]!='E')
78 
{
79
if(rec[0]=='A')
80 
{
81
scanf("%d %d",&i,&j);
82
add(i,j,1);
83 
}
84
else if(rec[0]=='S')
85 
{
86
scanf("%d %d",&i,&j);
87
sub(i,j,1);
88 
}
89
else if(rec[0]=='Q')
90 
{
91
scanf("%d %d",&i,&j);
92
printf("%dn",query(i,j+1,1));
93 
}
94 
}
95 
}
96
return 0;
97 }

 复习时重新写的代码

 1 #include <iostream>
 2 #include<bits/stdc++.h>
 3 #include <stack>
 4 #include <queue>
 5 #include <map>
 6 #include <set>
 7 #include <cstdio>
 8 #include <cstring>
 9 #include <algorithm>
10 #include <math.h>
11 using namespace std;
12 typedef long long ll;
13 typedef unsigned long long ull;
14 const int MAX=5e4+5;
15 int sum[MAX<<2];
16 void init(int l,int r,int k)
17 {
18
if(l==r)
19 
{
20
scanf("%d",&sum[k]);
21
return;
22 
}
23
int mid=l+r>>1;
24
init(l,mid,2*k);
25
init(mid+1,r,2*k+1);
26
sum[k]=sum[2*k]+sum[2*k+1];
27 }
28 void update(int lo,int num,int l,int r,int k)
29 {
30
if(l==r)
31 
{
32
sum[k]+=num;
33
return;
34 
}
35
int mid=l+r>>1;
36
if(lo<=mid)
37
update(lo,num,l,mid,2*k);
38
else
39
update(lo,num,mid+1,r,2*k+1);
40
sum[k]=sum[2*k]+sum[2*k+1];
41 }
42 int query(int ql,int qr,int l,int r,int k)
43 {
44
if(l>=ql&&r<=qr)
45
return sum[k];
46
int mid=l+r>>1;
47
if(qr<=mid)
48
return query(ql,qr,l,mid,2*k);
49
else if(ql>mid)
50
return query(ql,qr,mid+1,r,2*k+1);
51
else return query(ql,qr,l,mid,2*k)+query(ql,qr,mid+1,r,2*k+1);
52 }
53 int t,n;
54 char com[10];
55 int lo,num;
56 int main()
57 {
58
scanf("%d",&t);
59
for(int i=1;i<=t;i++)
60 
{
61
scanf("%d",&n);
62
init(1,n,1);
63
printf("Case %d:n",i);
64
while(~scanf("%s",com))
65 
{
66
if(com[0]=='E')
67
break;
68
else
69 
{
70
scanf("%d %d",&lo,&num);
71
if(com[0]=='A')
72
update(lo,num,1,n,1);
73
else if(com[0]=='S')
74
update(lo,-num,1,n,1);
75
else
76
printf("%dn",query(lo,num,1,n,1));
77 
}
78 
}
79 
}
80 }