1. 程式人生 > >A Simple Problem with Integers 【線段樹區間更新 區間查詢】

A Simple Problem with Integers 【線段樹區間更新 區間查詢】

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;
}