敌兵布阵
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 59891 Accepted Submission(s): 25331
Problem 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
分析:线段树计算区间和+点修改
代码:
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <ctype.h> #include <math.h> #include <queue> #include <iostream> #include <algorithm> #define INF 0x3f3f3f3f using namespace std; struct node { int ll, rr; int mm;//保存区间和 }q[200000+10]; int fa[50000+2]; /* 线段树是一种树状的数据结构,但在算法的实现过程中, 用的是一维结构体数组来模拟的, */ void Build(int i, int ll, int rr) { q[i].ll=ll; q[i].rr=rr;//当前线段节点的左右区间 q[i].mm=0; if(ll==rr){ fa[ll]=i; return; }//如果当前的是叶子节点 //fa[]记录下了每个叶子节点的存储位置 Build(i*2, ll, (ll+rr)/2); Build(i*2+1, (ll+rr)/2+1, rr); } //往下递归两个儿子节点 void update(int ri) { if(ri==1){ return ; }//到达根节点 修改完成返回 int fi=ri/2; q[fi].mm = q[fi*2].mm + q[fi*2+1].mm; update(ri/2); } int ans; void query(int i, int ll ,int rr) { if(q[i].ll==ll&&q[i].rr==rr){ ans=ans+q[i].mm; //printf("%d--\n", ans ); return; }// i=i*2; if(ll<=q[i].rr){//在这个子节点有范围 if(rr<=q[i].rr){ query(i, ll, rr); }else{ query(i, ll, q[i].rr); } } i=i+1; if(rr>=q[i].ll){ if(ll>=q[i].ll ){ query(i, ll, rr); }else{ query(i, q[i].ll, rr); } } } int main() { int tg; scanf("%d", &tg); int cnt=1; int n, u, v; char s[20]; while(tg--){ scanf("%d", &n); Build(1, 1, n);//建树 int cur; for(int i=1; i<=n; i++){ scanf("%d", &cur); q[fa[i]].mm=cur; update(fa[i]); } printf("Case %d:\n", cnt++); while(scanf("%s", s)!=EOF) { if(s[0]==‘E‘) break; else if(s[0]==‘Q‘){ scanf("%d %d", &u, &v); /* for(int i=1; i<=10; i++) printf("%d ", q[fa[i]].mm ); printf("\n\n"); */ ans=0; query(1, u, v); printf("%d\n", ans); } else if(s[0]==‘A‘){ scanf("%d %d", &u, &v); q[fa[u]].mm+=v; update(fa[u]); } else if(s[0]==‘S‘){ scanf("%d %d", &u, &v); q[fa[u]].mm-=v; update(fa[u]); } } } return 0; }
Sample Output
Case 1: 6 33 59