A Simple Problem with Integers 【線段樹區間更新 區間查詢】
阿新 • • 發佈:2018-12-10
POJ 3468
題意就不說了,之間看程式碼;
程式碼是對的,如果大佬覺得註釋有問題可以說,會仔細更改?
#include<stdio.h> #include<string.h> #include<string> #include<iostream> #include<algorithm> #include<queue> #include<math.h> #include<map> #include<vector> #include<stack> #define inf 500005 using namespace std; typedef long long ll; const int N=800055; ll tree[N],add[N]; ll m,n; void build(ll a,ll b,ll r)//建樹 { if(a==b) { scanf("%lld",&tree[r]); return; } ll mid=(a+b)/2; build(a,mid,r<<1);//不斷往下推,直到推到底端葉子(葉子就是樹的最底端) build(mid+1,b,r<<1|1); tree[r]=tree[r<<1]+tree[r<<1|1];//節點是下面兩個節點之和 } void pushdown(ll r,ll lz,ll rz)//區間維護 { //lz是在r節點左邊的底端葉子數,rz代表是右邊的底端葉子數 if(add[r]) { add[r<<1]+=add[r]; add[r<<1|1]+=add[r]; tree[r<<1]+=add[r]*lz; tree[r<<1|1]+=add[r]*rz; add[r]=0; } } void update(ll a,ll b,ll r,ll qa,ll qb,ll c)//區間更新 { if(a>=qa&&b<=qb) { add[r]+=c; tree[r]+=(b-a+1)*c; return; } ll mid=(a+b)/2; pushdown(r,mid+1-a,b-mid); if(mid>=qa)//因為需要更改qa到qb這個區間,所以如果mid還大於qa就一直往左查 update(a,mid,r<<1,qa,qb,c); if(mid<qb)//mid<qb就一直往右查 update(mid+1,b,r<<1|1,qa,qb,c); tree[r]=tree[r<<1]+tree[r<<1|1]; } ll qurey(ll a,ll b,ll r,ll qa,ll qb)//區間查詢 { if(a>=qa&&b<=qb) return tree[r]; ll mid=(a+b)/2; pushdown(r,mid+1-a,b-mid);//每次都需要向下維護區間 ll ans=0; if(mid>=qa)//和上面的查詢一樣 ans+=qurey(a,mid,r<<1,qa,qb); if(mid<qb) ans+=qurey(mid+1,b,r<<1|1,qa,qb); return ans; } int main() { while(~scanf("%lld %lld",&n,&m)) { ll i,a,b,c; char s; build(1,n,1); for(i=0;i<m;i++) { getchar(); scanf("%c",&s); if(s=='C') { scanf("%lld %lld %lld",&a,&b,&c); update(1,n,1,a,b,c); } else { scanf("%lld %lld",&a,&b); ll ans=qurey(1,n,1,a,b); printf("%lld\n",ans); } } } return 0; }