Segment Tree線段樹

線段樹的作用范圍：區(qū)間單點更新和覆蓋，區(qū)間多點更新和覆蓋些己。
時間復(fù)雜度：O（logN）
原理圖：

線段樹

對于一個長度為N的數(shù)組钓丰，我們將一個區(qū)間（L_{R的區(qū)間）劃分成兩個單元區(qū)間炭剪。每個區(qū)域分為左區(qū)域和右區(qū)域，左區(qū)域的范圍為L}（L+R）/2先壕，右區(qū)域的范圍為（L+R）/2+1~R瘩扼。當(dāng)左區(qū)域等于右區(qū)域的時候，就說明走到最底層了垃僚。對于每個區(qū)域的下標(biāo)我們規(guī)定集绰，左區(qū)域的下標(biāo)=2父節(jié)點的下標(biāo)，右區(qū)域的下標(biāo)=2父節(jié)點的下標(biāo)+1冈在。

結(jié)構(gòu)體
struct node
{
    int tl, tr;
    int lazy, val;
} tree[4*maxn];

建樹
void Push(int id)
{
    tree[id].val=tree[id*2].val+tree[id*2+1].val;
}

void Build(int id, int tl, int tr)
{
    tree[id].tl=tl;
    tree[id].tr=tr;
    tree[id].val=0;
    tree[id].lazy=0;
    if(tl== tr) tree[id].val=a[tl];
    else
    {
        int tm=(tl+tr)/2;
        Build(id*2, tl, tm);
        Build(id*2+1, tm+1, tr);
        Push(id);
    }
}

更新
void Pushdown(int id)
{
    if(tree[id].lazy==0) return ;
    tree[id*2].lazy=tree[id*2+1].lazy=tree[id].lazy;
    tree[id*2].val=(tree[id*2].tr-tree[id*2].tl+1)*tree[id].lazy;
    tree[id*2+1].val=(tree[id*2+1].tr-tree[id*2+1].tl+1)*tree[id].lazy;
    tree[id].lazy=0;
}

void Update(int id, int ql, int qr, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql>tr || qr<tl) return ;
    if(ql<=tl && tr<=qr)
    {
        tree[id].lazy=val;
        tree[id].val=(tree[id].tr-tree[id].tl+1)*tree[id].lazy;
        return ;
    }
    Pushdown(id);
    Update(id*2, ql, qr, val);
    Update(id*2+1, ql, qr, val);
    Push(id);
}

查詢
int Query(int id, int ql, int qr)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    int ans=0;
    if(ql>tr || qr<tl) return 0;
    if(ql<=tl && tr<=qr) return tree[id].val;
    Pushdown(id);
    ans+=Query(id*2, ql, qr);
    ans+=Query(id*2+1, ql, qr);
    Push(id);
    return ans;
}

線段樹的單點更新

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1000005;
const int inf = 0x7fffffff;
int a[maxn];

struct node
{
    int tl, tr, val;
} tree[4*maxn];

void Push(int id)
{
    tree[id].val=min(tree[id*2].val, tree[id*2+1].val);
}

void Build(int id, int tl, int tr)
{
    tree[id].tl=tl;
    tree[id].tr=tr;
    if(tl==tr) tree[id].val=a[tl];
    else
    {
        int mid=(tl+tr)/2;
        Build(id*2, tl, mid);
        Build(id*2+1, mid+1, tr);
        Push(id);
    }
}

void Update(int id, int i, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(i==tl && i==tr) tree[id].val=val;
    else
    {
        int tm=(tl+tr)/2;
        if(tm>=i) Update(id*2, i, val);
        else Update(id*2+1, i, val);
        Push(id);
    }
}

int Query(int id, int ql, int qr)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql<=tl && tr<=qr) return tree[id].val;
    int tm=(tl+tr)/2;
    int res1=inf, res2=inf;
    if(tm>=ql) res1=Query(id*2, ql, qr);
    if(tm<qr) res2=Query(id*2+1, ql, qr);
    return min(res1, res2);
}

int main()
{
    int i, n, q, op, l, r, val;
    scanf("%d", &n);
    for(i=1; i<=n; i++)
    {
        scanf("%d", &a[i]);
    }
    scanf("%d", &q);
    Build(1, 1, n);
    while(q--)
    {
        scanf("%d", &op);
        if(op==0)
        {
            scanf("%d%d", &l, &r);
            printf("%d\n", Query(1, l, r));
        }
        else
        {
            scanf("%d%d", &i, &val);
            Update(1, i, val);
        }
    }
    return 0;
}

線段樹的區(qū)間更新

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1000005;
const int inf = 0x7fffffff;

int a[maxn];
struct node
{
    int tl, tr, val, lazy;
} tree[4*maxn];

void push(int id)
{
    tree[id].val=tree[id*2].val+tree[id*2+1].val;
}

void pushdown(int id)
{
    if(tree[id].lazy==0) return ;
    tree[id*2].lazy=tree[id*2+1].lazy=tree[id].lazy;
    tree[id*2].val=(tree[id*2].tr-tree[id*2].tl+1)*tree[id].lazy;
    tree[id*2+1].val=(tree[id*2+1].tr-tree[id*2+1].tl+1)*tree[id].lazy;
    tree[id].lazy=0;
}

void build(int id, int tl, int tr)
{
    tree[id].tl=tl;
    tree[id].tr=tr;
    tree[id].val=0;
    tree[id].lazy=0;
    if(tl==tr) tree[id].val=a[tl];
    else
    {
        int tm=(tl+tr)/2;
        build(id*2, tl, tm);
        build(id*2+1, tm+1, tr);
        push(id);
    }
}

void update(int id, int ql, int qr, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql>tr || qr<tl) return ;
    if(ql<=tl && tr<=qr)
    {
        tree[id].lazy=val;
        tree[id].val=(tree[id].tr-tree[id].tl+1)*tree[id].lazy;
        return ;
    }
    pushdown(id);
    update(id*2, ql, qr, val);
    update(id*2+1, ql, qr, val);
    push(id);
}

int query(int id, int ql, int qr)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    int ans=0;
    if(ql>tr || qr<tl) return 0;
    if(ql<=tl && tr<=qr) return tree[id].val;
    pushdown(id);
    ans+=query(id*2, ql, qr);
    ans+=query(id*2+1, ql, qr);
    push(id);
    return ans;
}

int main()
{
    int n, i, q, l, r, val, op;
    scanf("%d", &n);
    for(i=1; i<=n; i++)
    {
        scanf("%d", &a[i]);
    }
    build(1, 1, n);
    scanf("%d", &q);
    while(q--)
    {
        scanf("%d", &op);
        if(op==0)
        {
            scanf("%d%d", &l, &r);
            printf("%d\n", query(1, l, r));
        }
        else
        {
            scanf("%d%d%d", &l, &r, &val);
            update(1, l, r, val);
        }
    }
    return 0;
}

離散化

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1000005;
const int inf = 0x7fffffff;

int vis[maxn];
int ans=0;
int a[maxn];
struct re
{
    int l, r;
} q[maxn];

struct node
{
    int tl, tr, val, lazy;
} tree[4*maxn];

void Build(int id, int tl, int tr)
{
    tree[id].tl=tl;
    tree[id].tr=tr;
    tree[id].val=0;
    tree[id].lazy=0;
    if(tl+1==tr) return ;
    else
    {
        int tm=(tl+tr)/2;
        Build(id*2, tl, tm);
        Build(id*2+1, tm, tr);
    }
}

void Pushdown(int id)
{
    if(tree[id].lazy==0) return ;
    tree[id*2].lazy=tree[id*2+1].lazy=tree[id].lazy;
    tree[id*2].val=tree[id*2+1].val=tree[id].lazy;
    tree[id].lazy=0;
}

void Update(int id, int ql, int qr, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql<=tl && tr<=qr)
    {
        tree[id].lazy=val;
        tree[id].val=val;
        return ;
    }
    Pushdown(id);
    int tm=(tl+tr)/2;
    if(tm>ql) Update(id*2, ql, qr, val);
    if(tm<qr) Update(id*2+1, ql, qr, val);
}

void Query(int id, int ql, int qr)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(tl+1==tr)
    {
        if(vis[tree[id].val]==0)
        {
            vis[tree[id].val]=1;
            ans++;
        }
        return ;
    }
    Pushdown(id);
    int tm=(tl+tr)/2;
    if(tm>ql) Query(id*2, ql, qr);
    if(tm<qr) Query(id*2+1, ql, qr);
}

void init()
{
    memset(vis,0,sizeof(vis));
    memset(a,0,sizeof(a));
    ans=0;
}

int main()
{
    init();
    int i, n, l, k=0;
    scanf("%d%d", &n, &l);
    for(i=1; i<=n; i++)
    {
        scanf("%d%d", &q[i].l, &q[i].r);
        a[k++]=q[i].l;
        a[k++]=q[i].r;
    }
    sort(a, a+k);
    int len=unique(a, a+k)-a;
    Build(1, 1, len);
    for(i=1; i<=n; i++)
    {
        int l=lower_bound(a,a+len,q[i].l)-a+1;
        int r=lower_bound(a,a+len,q[i].r)-a+1;
        Update(1, l, r, i);
    }
    vis[0]=1;
    Query(1, 1, len);
    printf("%d\n", ans);
    return 0;
}

雙標(biāo)記線段樹

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1e5+5;
const int inf = 0x7fffffff;

struct node
{
    int tl, tr;
    int lazyset, lazyadd, val;
} tree[4*maxn];
int a[maxn];

void push(int id)
{
    tree[id].val=tree[2*id].val+tree[2*id+1].val;
}

void build(int id, int tl, int tr)
{
    tree[id].tl=tl;
    tree[id].tr=tr;
    if(tl==tr) tree[id].val=a[tl];
    else
    {
        int tm=(tr+tl)/2;
        build(id*2,tl,tm);
        build(id*2+1,tm+1,tr);
        push(id);
    }
}

void pushdown(int id)
{
    if(tree[id].lazyset!=0)
    {
        tree[id*2].lazyset=tree[id].lazyset;
        tree[id*2+1].lazyset=tree[id].lazyset;
        tree[id*2].lazyadd=tree[id].lazyadd;
        tree[id*2+1].lazyadd=tree[id].lazyadd;
        int ans=tree[id].lazyset+tree[id].lazyadd;
        tree[id*2].val=(tree[id*2].tr-tree[id*2].tl+1)*ans;
        tree[id*2+1].val=(tree[id*2+1].tr-tree[id*2+1].tl+1)*ans;
        tree[id].lazyset=0;
        tree[id].lazyadd=0;
    }
    else if(tree[id].lazyadd!=0)
    {
        tree[id*2].lazyadd+=tree[id].lazyadd;
        tree[id*2+1].lazyadd+=tree[id].lazyadd;
        tree[id*2].val+=(tree[id*2].tr-tree[id*2].tl+1)*tree[id].lazyadd;
        tree[id*2+1].val+=(tree[id*2+1].tr-tree[id*2+1].tl+1)*tree[id].lazyadd;
        tree[id].lazyadd=0;
    }
}

void putset(int id, int val)
{
    tree[id].lazyadd=0;
    tree[id].lazyset=val;
    tree[id].val=(tree[id].tr-tree[id].tl+1)*val;
}

void Set(int id, int ql, int qr, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql>tr || qr<tl) return ;
    if(ql<=tl && tr<=qr)
    {
        putset(id, val);
        return ;
    }
    pushdown(id);
    Set(id*2, ql, qr, val);
    Set(id*2+1, ql, qr, val);
    push(id);
}

void putadd(int id, int val)
{
    tree[id].lazyadd+=val;
    tree[id].val+=(tree[id].tr-tree[id].tl+1)*val;
}

void add(int id, int ql, int qr, int val)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    if(ql>tr || qr<tl) return ;
    if(ql<=tl && tr<=qr)
    {
        putadd(id, val);
        return ;
    }
    pushdown(id);
    add(id*2, ql, qr, val);
    add(id*2+1, ql, qr, val);
    push(id);
}

int getval(int id, int ql, int qr)
{
    int tl=tree[id].tl;
    int tr=tree[id].tr;
    int ans=0;
    if(ql>tr || qr<tl) return 0;
    if(ql<=tl && tr<=qr) return tree[id].val;

    pushdown(id);
    ans+=getval(id*2, ql, qr);
    ans+=getval(id*2+1, ql, qr);
    push(id);
    return ans;
}

int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=1; i<=n+1; i++)
        scanf("%d",&a[i]);
    build(1,1,n+1);
    while(q--)
    {
        int op,l,r,val;
        scanf("%d%d%d%d",&op,&l,&r,&val);
        if(!op)add(1,l+1,r+1,val);
        else Set(1,l+1,r+1,val);
        printf("%d\n",getval(1,1,n+1));
    }
    return 0;
}

最后編輯于：2017.12.09 02:20:57

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者

人面猴
序言：七十年代末倒慧，一起剝皮案震驚了整個濱河市，隨后出現(xiàn)的幾起案子包券，更是在濱河造成了極大的恐慌纫谅，老刑警劉巖，帶你破解...
沈念sama閱讀 221,695評論 6贊 515
死咒
序言：濱河連續(xù)發(fā)生了三起死亡事件溅固，死亡現(xiàn)場離奇詭異付秕，居然都是意外死亡，警方通過查閱死者的電腦和手機(jī)侍郭，發(fā)現(xiàn)死者居然都...
沈念sama閱讀 94,569評論 3贊 399
救了他兩次的神仙讓他今天三更去死
文/潘曉璐我一進(jìn)店門询吴，熙熙樓的掌柜王于貴愁眉苦臉地迎上來，“玉大人亮元，你說我怎么就攤上這事猛计。” “怎么了爆捞？”我有些...
開封第一講書人閱讀 168,130評論 0贊 360
道士緝兇錄：失蹤的賣姜人
文/不壞的土叔我叫張陵奉瘤，是天一觀的道長。經(jīng)常有香客問我煮甥，道長盗温，這世上最難降的妖魔是什么？我笑而不...
開封第一講書人閱讀 59,648評論 1贊 297
?港島之戀（遺憾婚禮）
正文為了忘掉前任成肘，我火速辦了婚禮卖局，結(jié)果婚禮上，老公的妹妹穿的比我還像新娘双霍。我一直安慰自己砚偶，他們只是感情好批销，可當(dāng)我...
茶點故事閱讀 68,655評論 6贊 397
惡毒庶女頂嫁案：這布局不是一般人想出來的
文/花漫我一把揭開白布。她就那樣靜靜地躺著蟹演，像睡著了一般风钻。火紅的嫁衣襯著肌膚如雪。梳的紋絲不亂的頭發(fā)上酒请，一...
開封第一講書人閱讀 52,268評論 1贊 309
城市分裂傳說
那天骡技，我揣著相機(jī)與錄音，去河邊找鬼羞反。笑死布朦，一個胖子當(dāng)著我的面吹牛，可吹牛的內(nèi)容都是我干的昼窗。我是一名探鬼主播是趴，決...
沈念sama閱讀 40,835評論 3贊 421
雙鴛鴦連環(huán)套：你想象不到人心有多黑
文/蒼蘭香墨我猛地睜開眼，長吁一口氣：“原來是場噩夢啊……” “哼澄惊！你這毒婦竟也來了唆途？” 一聲冷哼從身側(cè)響起，我...
開封第一講書人閱讀 39,740評論 0贊 276
萬榮殺人案實錄
序言：老撾萬榮一對情侶失蹤掸驱，失蹤者是張志新（化名）和其女友劉穎肛搬，沒想到半個月后，有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體毕贼，經(jīng)...
沈念sama閱讀 46,286評論 1贊 318
?護(hù)林員之死
正文獨居荒郊野嶺守林人離奇死亡温赔，尸身上長有42處帶血的膿包…… 初始之章·張勛以下內(nèi)容為張勛視角年9月15日...
茶點故事閱讀 38,375評論 3贊 340
?白月光啟示錄
正文我和宋清朗相戀三年，在試婚紗的時候發(fā)現(xiàn)自己被綠了鬼癣。大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片陶贼。...
茶點故事閱讀 40,505評論 1贊 352
活死人
序言：一個原本活蹦亂跳的男人離奇死亡，死狀恐怖待秃，靈堂內(nèi)的尸體忽然破棺而出拜秧，到底是詐尸還是另有隱情，我是刑警寧澤章郁，帶...
沈念sama閱讀 36,185評論 5贊 350
?日本核電站爆炸內(nèi)幕
正文年R本政府宣布腹纳，位于F島的核電站，受9級特大地震影響驱犹，放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜足画，卻給世界環(huán)境...
茶點故事閱讀 41,873評論 3贊 333
男人毒藥：我在死后第九天來索命
文/蒙蒙一雄驹、第九天我趴在偏房一處隱蔽的房頂上張望。院中可真熱鬧淹辞，春花似錦医舆、人聲如沸。這莊子的主人今日做“春日...
開封第一講書人閱讀 32,357評論 0贊 24
一樁弒父案蔬将，背后竟有這般陰謀
文/蒼蘭香墨我抬頭看了看天上的太陽爷速。三九已至，卻和暖如春霞怀，著一層夾襖步出監(jiān)牢的瞬間惫东，已是汗流浹背。一陣腳步聲響...
開封第一講書人閱讀 33,466評論 1贊 272
情欲美人皮
我被黑心中介騙來泰國打工毙石，沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留廉沮，地道東北人。一個月前我還...
沈念sama閱讀 48,921評論 3贊 376
代替公主和親
正文我出身青樓徐矩，卻偏偏與公主長得像滞时，于是被迫代替她去往敵國和親。傳聞我的和親對象是個殘疾皇子滤灯，可洞房花燭夜當(dāng)晚...
茶點故事閱讀 45,515評論 2贊 359

Segment Tree線段樹

推薦閱讀更多精彩內(nèi)容