HDU1166 敌兵布阵 [线段树模板]
生活随笔
收集整理的這篇文章主要介紹了
HDU1166 敌兵布阵 [线段树模板]
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題意:在序列中修改單點(diǎn)和查詢區(qū)間和
#include<iostream> #include<cstdio> #include<cstring> #define ls l,m,rt<<1 #define rs m+1,r,rt<<1|1 using namespace std; const int maxn=50005*4; int sum[maxn<<2],add[maxn<<2];//sum求和,add懶惰標(biāo)記 int a[maxn]={0},n=50005;//存原數(shù)組數(shù)據(jù)下標(biāo)[1,n] //更新節(jié)點(diǎn)信息,這里是求和 void pushup(int rt) {sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } //建樹 void pushdown(int rt,int ln,int rn) {//ln,rn為左子樹,右子樹的數(shù)量if(add[rt])//下推標(biāo)記 {add[rt<<1]+=add[rt];add[rt<<1|1]+=add[rt];sum[rt<<1]+=add[rt]*ln;sum[rt<<1|1]+=add[rt]*rn;add[rt]=0;//清除本節(jié)點(diǎn)標(biāo)記 } } void build(int l,int r,int rt)//rt表示當(dāng)前節(jié)點(diǎn)編號(hào) {if(l==r){sum[rt]=a[l];return;}int m=(l+r)>>1;build(l,m,rt<<1);build(m+1,r,rt<<1|1);pushup(rt); }//點(diǎn)修改a[L]+=c; //比如update(a,-b,1,n,1);//修改a點(diǎn)的值,減去b,當(dāng)前節(jié)點(diǎn)區(qū)間是1到n,起始點(diǎn)是1 void update(int L,int c,int l,int r,int rt) {//l,r當(dāng)前節(jié)點(diǎn)區(qū)間,rt當(dāng)前節(jié)點(diǎn)編號(hào),L需要修改的節(jié)點(diǎn),c修改的值if(l==r){sum[rt]+=c;return;}int m=(l+r)>>1;//根據(jù)條件判斷調(diào)用左子樹還是右子樹if(L<=m)update(L,c,l,m,rt<<1);elseupdate(L,c,m+1,r,rt<<1|1);pushup(rt);//子節(jié)點(diǎn)更新,所以本節(jié)點(diǎn)也需要更新 }//區(qū)間修改a[r,t]+=c ////比如query(a,b,1,n,1))查詢a b之間的和,當(dāng)前節(jié)點(diǎn)區(qū)間是1到n,起始點(diǎn)是1 void update_(int L,int R,int c,int l,int r,int rt) {//L,R表示操作區(qū)間,l,r表示當(dāng)前節(jié)點(diǎn)區(qū)間,rt表示當(dāng)前節(jié)點(diǎn)編號(hào)if(L<=l&&r<=R)//遍歷的區(qū)間在操作區(qū)間內(nèi) {sum[rt]+=c*(r-l+1);add[rt]+=c;return;}int m=(l+r)>>1;pushdown(rt,m-l+1,r-m);if(L<=m)update_(L,R,c,l,m,rt<<1);if(R>m)update_(L,R,c,m+1,r,rt<<1|1);pushup(rt); } //區(qū)間查詢 a[l,r]的和 //下推標(biāo)記函數(shù)int query(int L,int R,int l,int r,int rt) {if(L<=l&&r<=R){return sum[rt];}int m=(l+r)>>1;pushdown(rt,m-l+1,r-m);int ans=0;if(L<=m)ans+=query(L,R,l,m,rt<<1);if(R>m)ans+=query(L,R,m+1,r,rt<<1|1);return ans; }int main() {int t,cnt=1;scanf("%d",&t);while(t--){memset(a,0,sizeof(a));memset(sum,0,sizeof(sum));memset(add,0,sizeof(add));scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%d",&a[i]);build(1,n,1);printf("Case %d:\n",cnt++);char str[20];while(scanf("%s",str)&&str[0]!='E'){int a,b;scanf("%d%d",&a,&b);if(str[0]=='Q'){query(a,b,1,n,1);}if(str[0]=='S')update(a,-b,1,n,1);//修改a點(diǎn)的值,減去b,當(dāng)前節(jié)點(diǎn)區(qū)間是1到n,起始點(diǎn)是1if(str[0]=='A')update(a,b,1,n,1);if(str[0]=='Q')printf("%d\n",query(a,b,1,n,1));//查詢a b之間的和,當(dāng)前節(jié)點(diǎn)區(qū)間是1到n,起始點(diǎn)是1 }}return 0; }?
轉(zhuǎn)載于:https://www.cnblogs.com/fudanxi/p/11172549.html
與50位技術(shù)專家面對(duì)面20年技術(shù)見證,附贈(zèng)技術(shù)全景圖總結(jié)
以上是生活随笔為你收集整理的HDU1166 敌兵布阵 [线段树模板]的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: mac php Swoole入门
- 下一篇: 简单七步让你轻松安全迁移网站