敌兵布阵(线段树单点更新+区间查询)
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的责骂的.
中央情报局要研究敌人究竟演习什么战术,所以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
#include <iostream> #include <algorithm> #include <cstdio> #include <string> #include <cstring> #include <cstdlib> #include <map> #include <vector> #include <set> #include <queue> #include <stack> #include <cmath> using namespace std; #define pq priority_queue<int> #define pql priority_queue<ll> #define pqn priority_queue<node> #define v vector<int> #define vl vector<ll> #define lson rt<<1, l, m #define rson rt<<1|1, m+1, r #define read(x) scanf("%d",&x) #define lread(x) scanf("%lld",&x); #define pt(x) printf("%d\n",(x)) #define yes printf("YES\n"); #define no printf("NO\n"); #define gcd __gcd #define out(x) cout<<x<<endl; #define over cout<<endl; #define rep(j,k) for (int i = (int)(j); i <= (int)(k); i++) #define input(k) for (int i = 1; i <= (int)(k); i++) {cin>>a[i] ; } #define mem(s,t) memset(s,t,sizeof(s)) #define re return ; #define ok return 0; #define TLE std::ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL); #define mod(x) ((x)%9973) #define test cout<<" ++++++ "<<endl; #define ls now<<1 #define rs now<<1|1 typedef long long ll; const int N=50000+5; const int maxn=10000+5; const int len = 2*1e5+5; typedef struct node { int val,l,r,sum,mx,mn,lazy; int mid() {return (l+r)>>1;} }node; node dp[len]; int a[N]; //double getlen(node xx,node yy) { return ( (xx.x-yy.x)*(xx.x-yy.x) +(xx.y-yy.y)*(xx.y-yy.y) ); } void Newmax(int now) { dp[now].mx = max( dp[ls].mx ,dp[rs].mx ); dp[now].mn = min( dp[ls].mn ,dp[rs].mn ); } void querymax(int now,int l,int r) { } void uplazy(int now) { if(dp[now].lazy) { dp[ls].sum+=dp[now].lazy*(dp[ls].r-dp[ls].l+1); dp[rs].sum+=dp[now].lazy*(dp[rs].r-dp[rs].l+1); dp[ls].lazy+=dp[now].lazy; dp[rs].lazy+=dp[now].lazy; dp[now].lazy = 0 ; } } int querysum(int now,int l,int r) { //test; if(dp[now].l>=l&&dp[now].r<=r) {return dp[now].sum;} uplazy(now); int cnt=0; if(dp[now].mid()>=r) cnt+=querysum(ls,l,r); else if(dp[now].mid()<l) cnt+=querysum(rs,l,r); else { //cout<<dp[now].l<<" "<<dp[now].r<<" "<<dp[now].mid()<<endl; int mid =(dp[now].l+dp[now].r)>>1 ; cnt+=querysum(ls,l,mid); cnt+=querysum(rs,mid+1,r); } //cout<<"cnt "<<cnt<<endl; return cnt; } void upsum(int now) { dp[now].sum = dp[ls].sum+dp[rs].sum; } void build(int now, int l,int r) { dp[now].l=l;dp[now].r=r;dp[now].lazy=0;dp[now].sum=0; if(l==r) { //cout<<"now "<<r<<" "<<dp[now].sum <<endl; dp[now].sum = a[l]; re; } int mid=(l+r)>>1; build(ls,l,mid); build(rs,1+mid,r); upsum(now); } //区间更新 void updata(int now,int l,int r,int ans ) { if(dp[now].l>=1 && dp[now].r<=r) { dp[now].sum+=ans*(dp[now].r-dp[now].l+1); re; } uplazy(now); if(dp[now].mid()>=r) updata(ls,l,r,ans); else if(dp[now].mid()+1<=l) updata(rs,l,r,ans); else { int mid = dp[now].mid(); updata(ls,l,mid,ans); updata(rs,mid+1,r,ans); } upsum(now); } void upnode(int now,int l,int k) { if(dp[now].l==dp[now].r) { dp[now].sum+=k; re; } int mid =(dp[now].l+dp[now].r)>>1; if(l<=mid) upnode(ls,l,k); else upnode(rs,l,k); dp[now].sum = dp[ls].sum+dp[rs].sum; } int main() { TLE; int t,ll,rr,n; string str; cin>>t; for(int kk=1;kk<=t;kk++) { cout<<"Case "<<kk<<":"<<endl; cin>>n; input(n); build(1,1,n); //test; while(cin>>str&&str[0]!='E') { if(str[0]=='Q') { cin>>ll>>rr; cout<<querysum(1,ll,rr)<<endl; } else if(str[0]=='A') { cin>>ll>>rr; //updata(1,ll,rr,ans); upnode(1,ll,rr); } else if(str[0]=='S') { cin>>ll>>rr; upnode(1,ll,-1*rr); } } } ok; }
所遇皆星河