首页 > 代码库 > 敌兵布阵(线段树)
敌兵布阵(线段树)
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的责骂的.
接下来每行有一条命令,命令有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条命令
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
Case 1: 6 33 59
#include<iostream> #include<cstring> #include<cstdio> #define MX 50005 using namespace std; struct node { int l, r, v; } a[MX << 2]; void PhosUp(int root) { a[root].v = a[root << 1].v + a[root << 1 | 1].v; } void build_tree(int l, int r, int root) { a[root].l = l; a[root].r = r; a[root].v = 0; if (l == r) { scanf("%d", &a[root].v); return ; } int mid = (l + r) >> 1; build_tree(l, mid, root << 1); build_tree(mid + 1, r, root << 1 | 1); PhosUp(root); } void update(int l, int r, int root, int k) { if (a[root].l == l & a[root].r == r) { a[root].v += k; return ; } int mid = (a[root].l + a[root].r) >> 1; if (r <= mid)update(l, r, root << 1, k); else if (l > mid)update(l, r, root << 1 | 1, k); else { update(l, mid, root << 1, k); update(mid + 1, r, root << 1 | 1, k); } PhosUp(root); } int Query(int l, int r, int root) { if (a[root].l == l && a[root].r == r) { return a[root].v; } int mid = (a[root].l + a[root].r) >> 1; int ans = 0; if (r <= mid)ans += Query(l, r, root << 1); else if (l > mid)ans += Query(l, r, root << 1 | 1); else ans += Query(l, mid, root << 1) + Query(mid + 1, r, root << 1 | 1); return ans; } int main() { int t, c = 1, n, l, r; char ch[10]; scanf("%d", &t); while (t--) { scanf("%d", &n); printf("Case %d:\n", c++); build_tree(1, n, 1); while (1) { scanf("%s", ch); if (ch[0] == 'E') { break; } if (ch[0] == 'Q') { scanf("%d%d", &l, &r); printf("%d\n", Query(l, r, 1)); } if (ch[0] == 'A') { scanf("%d%d", &l, &r); update(l, l, 1, r); } if (ch[0] == 'S') { scanf("%d%d", &l, &r); update(l, l, 1, -r); } } } return 0; }
敌兵布阵(线段树)