線段樹 SegTree

created by Dejavu
(不斷更新中)


  • 概念

線段樹唾那,類似區(qū)間樹觉壶,是一個(gè)完全二叉樹官觅,它在各個(gè)節(jié)點(diǎn)保存一條線段(數(shù)組中的一段子數(shù)組)孩等,主要用于高效解決連續(xù)區(qū)間的動(dòng)態(tài)查詢問題,由于二叉結(jié)構(gòu)的特性署海,它基本能保持每個(gè)操作的復(fù)雜度為O(lgN)吗购,你可以在數(shù)據(jù)結(jié)構(gòu)可視(visualgo)網(wǎng)站中看到算法的具體搭建過程。

  • c++ 模板代碼

這里的代碼是對(duì)區(qū)間寫入優(yōu)化后的代碼砸狞,加入了lazy_tag用來標(biāo)記區(qū)間更新的位置

//最好不要用頭文件bits/stdc++.h 因?yàn)橐恍┚幾g器會(huì)不支持這個(gè)頭文件

#include <iostream>
using namespace std;
#define MaxN 50005
class SegTree {
public:
    int left, right;
    int sum;
    int lazy_tag;
    int mid() { return (left + right) >> 1; }
};
SegTree tree[MaxN];
void pushUp(int rt) { tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum; }
void pushDown(int rt, int len) {
    int tag = tree[rt].lazy_tag;
    if (tag) {
        tree[rt << 1].lazy_tag += tag;
        tree[rt << 1 | 1].lazy_tag += tag;
        tree[rt << 1].sum += tag*(len - (len >> 1));
        tree[rt << 1 | 1].sum += tag*(len >> 1);
        tree[rt].lazy_tag = 0;
    }
}
void build(int rt, int left, int right) {
    tree[rt].left = left;
    tree[rt].right = right;
    tree[rt].lazy_tag = 0;
    if (left == right) { cin >> tree[rt].sum;return; }
    int mid = tree[rt].mid();
    build(rt << 1, left, mid);
    build(rt << 1 | 1, mid + 1, right);
    pushUp(rt);
}
void update(int rt, int val, int left, int right) {
    if (tree[rt].left == left && tree[rt].right == right) {
        tree[rt].lazy_tag += val;
        tree[rt].sum += val*(right - left + 1);
        return;
    }
    if (tree[rt].left == tree[rt].right) return;
    pushDown(rt, tree[rt].right - tree[rt].left + 1);
    int mid = tree[rt].mid();
    if (right <= mid) update(rt << 1, val, left, right);
    else if (left > mid) update(rt << 1 | 1, val, left, right);
    else {
        update(rt << 1, val, left, mid);
        update(rt << 1 | 1, val, mid + 1, right);
    }
    pushUp(rt);
}
int query(int rt, int left, int right) {
    if (tree[rt].left == left && tree[rt].right == right) return tree[rt].sum;
    pushDown(rt, tree[rt].right - tree[rt].left + 1);
    int mid = tree[rt].mid();
    int sum(0);
    if (right <= mid) sum += query(rt << 1, left, right);
    else if (left > mid) sum += query(rt << 1 | 1, left, right);
    else {
        sum += query(rt << 1, left, mid);
        sum += query(rt << 1 | 1, mid + 1, right);
    }
    return sum;
}
int main() {
    ios::sync_with_stdio(false);
    int n;cin >> n;
    build(1, 1, n);
    char cmd;
    int in1, in2, val;
    while (cin >> cmd, cmd != 'E') {
        if (cmd == 'Q') { cin >> in1 >> in2;cout << query(1, in1, in2) << endl; }
        if (cmd == 'A') { cin >> in1 >> in2 >> val;update(1, val, in1, in2); }
    }
}

例題

題解和代碼性能分析

//c++ 用類似c的方法寫
#include <stdio.h>
#include <string.h>
#define MaxN 50000
int t[MaxN * 4];
inline void pushUp(int rt) { t[rt] = t[rt << 1] + t[rt << 1 | 1]; }
void build(int rt,int l,int r) {
    if (l == r) {scanf("%d",&t[rt]);return;}
    int m = (l + r) >> 1;
    build(rt << 1, l, m);
    build(rt << 1 | 1, m + 1, r);
    pushUp(rt);
}
void update(int rt, int l, int r, int p, int v) {
    if (l == r) { t[rt] += v;return; }
    int m = (l + r) >> 1;
    if (p <= m) update(rt << 1, l, m, p, v);
    else update(rt << 1 | 1, m + 1, r, p, v);
    pushUp(rt);
}
int query(int rt,int l,int r, int _l, int _r) {
    if (_l <= l && r <= _r) return t[rt];
    int m = (l + r) >> 1;
    int sum(0);
    if (_r <= m) sum += query(rt << 1, l, m, _l, _r);
    else if (_l > m) sum += query(rt << 1 | 1, m + 1, r, _l, _r);
    else {
        sum += query(rt << 1, l, m, _l, _r);
        sum += query(rt << 1 | 1, m + 1, r, _l, _r);
    }
    return sum;
}
void main() {
    int n, tn;scanf("%d",&n);tn = n;
    while (tn--) {
        int num;scanf("%d", &num);
        build(1, 1, num);
        printf("Case %d:\n", n - tn);
        char cmd[10];
        int in1, in2;
        while (scanf("%s", &cmd)) {
            if (cmd[0] == 'E') break;
            scanf("%d%d", &in1, &in2);
            if (cmd[0] == 'A') update(1, 1, num, in1, in2);
            else if (cmd[0] == 'S') update(1, 1, num, in1, -in2);
            else printf("%d\n", query(1, 1, num, in1, in2));
            memset(cmd, '\0', sizeof(cmd));
        }
    }
}
上面代碼在hdu運(yùn)行的結(jié)果
//更改query函數(shù)
int sum(0);
void query(int rt, int l, int r, int _l, int _r) {
    if (_l <= l && r <= _r) {sum += t[rt];return;}
    int m = (l + r) >> 1;
    if (_r <= m) query(rt << 1, l, m, _l, _r);
    else if (_l > m) query(rt << 1 | 1, m + 1, r, _l, _r);
    else {
        query(rt << 1, l, m, _l, _r);
        query(rt << 1 | 1, m + 1, r, _l, _r);
    }
}
更改query函數(shù)將sum作為全局變量
//c++ 用c寫法下捻勉,全宏定義
#include <stdio.h>
#include <string.h>
#define MaxN 50000
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
#define mid int m=(l+r)>>1
#define base int rt,int l,int r
#define we_base 1,1,num
int t[MaxN*4];
inline void pushUp(int rt) { t[rt] = t[rt << 1] + t[rt << 1 | 1]; }
void build(base) {
    if (l == r) { scanf("%d", &t[rt]);return; }
    mid;
    build(lson);
    build(rson);
    pushUp(rt);
}
void update(base, int p, int v) {
    if (l == r) { t[rt] += v;return; }
    mid;
    if (p <= m) update(lson, p, v);
    else update(rson, p, v);
    pushUp(rt);
}
int query(base, int _l, int _r) {
    if (_l <= l && r <= _r) return t[rt];
    mid;
    int sum(0);
    if (_r <= m) sum += query(lson, _l, _r);
    else if (_l > m) sum += query(rson, _l, _r);
    else {
        sum += query(lson, _l, _r);
        sum += query(rson, _l, _r);
    }
    return sum;
}
void main() {
    int n, tn;scanf("%d", &n);tn = n;
    while (tn--) {
        int num;scanf("%d", &num);
        build(we_base);
        printf("Case %d:\n", n - tn);
        char cmd[10];
        int in1, in2;
        while (scanf("%s", &cmd)) {
            if (cmd[0] == 'E') break;
            scanf("%d%d", &in1, &in2);
            if (cmd[0] == 'A') update(we_base, in1, in2);
            else if (cmd[0] == 'S') update(we_base, in1, -in2);
            else printf("%d\n", query(we_base, in1, in2));
            memset(cmd, '\0', sizeof(cmd));
        }
    }
}
c寫法全宏定義

可見使用宏定義的算法運(yùn)行速度比普通寫法稍快一些,而且寫代碼的速度將會(huì)比普通寫法快



//用c++的形式寫
#include <iostream>
#include <string>
using namespace std;
#define MaxN 50000
int t[MaxN * 4];
inline void pushUp(int rt) { t[rt] = t[rt << 1] + t[rt << 1 | 1]; }
void build(int rt,int l,int r) {
    if (l == r) { cin >> t[rt];return; }
    int m = (l + r) >> 1;
    build(rt << 1, l, m);
    build(rt << 1 | 1, m + 1, r);
    pushUp(rt);
}
void update(int rt, int l, int r, int p, int v) {
    if (l == r) { t[rt] += v;return; }
    int m = (l + r) >> 1;
    if (p <= m) update(rt << 1, l, m, p, v);
    else update(rt << 1 | 1, m + 1, r, p, v);
    pushUp(rt);
}
int query(int rt,int l,int r, int _l, int _r) {
    if (_l <= l && r <= _r) return t[rt];
    int m = (l + r) >> 1;
    int sum(0);
    if (_r <= m) sum += query(rt << 1, l, m, _l, _r);
    else if (_l > m) sum += query(rt << 1 | 1, m + 1, r, _l, _r);
    else {
        sum += query(rt << 1, l, m, _l, _r);
        sum += query(rt << 1 | 1, m + 1, r, _l, _r);
    }
    return sum;
}
void main() {
    ios::sync_with_stdio(false);
    int n, tn;cin >> n;tn = n;
    while (tn--) {
        int num;cin >> num;
        build(1, 1, num);
        cout << "Case " << n - tn <<":" << endl;
        string cmd;
        int in1, in2;
        while (cin >> cmd, cmd != "End") {
            cin>> in1 >> in2;
            if (cmd == "Add") update(1, 1, num, in1, in2);
            else if (cmd == "Sub") update(1, 1, num, in1, -in2);
            else cout << query(1, 1, num, in1, in2) << endl;
        }
    }
}

ios::sync_with_stdio(false)

ios::sync_with_stdio(true)

這里的1000ms并不是真正的運(yùn)行時(shí)間而是超時(shí)后就停止運(yùn)行了刀森,就是說當(dāng)設(shè)置為true時(shí)踱启,運(yùn)行時(shí)間是遠(yuǎn)遠(yuǎn)超過設(shè)置為false時(shí)的時(shí)間的,sync_with_stdio(bool)這個(gè)語句是詢問是否與stdio庫同步,同步過程會(huì)耗費(fèi)很多時(shí)間

//c++ 下全宏定義寫法
#include <iostream>
#include <string>
using namespace std;
#define MaxN 50000
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
#define mid int m=(l+r)>>1
#define base int rt,int l,int r
#define we_base 1,1,num
int t[MaxN*4];
inline void pushUp(int rt) { t[rt] = t[rt << 1] + t[rt << 1 | 1]; }
void build(base) {
    if (l == r) { cin >> t[rt];return; }
    mid;
    build(lson);
    build(rson);
    pushUp(rt);
}
void update(base, int p, int v) {
    if (l == r) { t[rt] += v;return; }
    mid;
    if (p <= m) update(lson, p, v);
    else update(rson, p, v);
    pushUp(rt);
}
int query(base, int _l, int _r) {
    if (_l <= l && r <= _r) return t[rt];
    mid;
    int sum(0);
    if (_r <= m) sum += query(lson, _l, _r);
    else if (_l > m) sum += query(rson, _l, _r);
    else {
        sum += query(lson, _l, _r);
        sum += query(rson, _l, _r);
    }
    return sum;
}
void main() {
    ios::sync_with_stdio(false);
    int n, tn;cin >> n;tn = n;
    while (tn--) {
        int num;cin >> num;
        build(we_base);
        cout << "Case " << n - tn << ":" << endl;
        string cmd;
        int in1, in2;
        while (cin >> cmd, cmd != "End") {
            cin >> in1 >> in2;
            if (cmd == "Add") update(we_base, in1, in2);
            else if (cmd == "Sub") update(we_base, in1, -in2);
            else cout << query(we_base, in1, in2) << endl;
        }
    }
}
c++ 下全宏定義寫法

然而使用c++下的頭文件和空間名埠偿,似乎用宏定義效果并不好

代碼運(yùn)行各項(xiàng)性能對(duì)比

c寫法


c++下普通c的寫法

更改query函數(shù)將sum作為全局變量

c寫法全宏定義

c++寫法


ios::sync_with_stdio(false)

ios::sync_with_stdio(true)

c++ 下全宏定義寫法

用c++的輸入輸出流在性能上會(huì)比c的輸入輸出流慢一些透罢,即使加了去同步語句,所以acm的代碼最好采用c的輸入輸出流

#include <cstdio>
#define s scanf
#define p printf
#define base int rt,int l,int r
#define we_base 1,1,num
#define max(v1,v2) v1>v2?v1:v2
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
#define mid int m=(l+r)>>1
#define MaxN 200000
int t[MaxN * 4];
inline void pushUp(int rt) { t[rt] = max(t[rt << 1], t[rt << 1 | 1]); }
void build(base) {
    if (l == r) s("%d", &t[rt]);
    else {
        mid;
        build(lson);
        build(rson);
        pushUp(rt);
    }
}
void update(base, int pos, int v) {
    if (l == r) t[rt] = v;
    else {
        mid;
        if (pos <= m) update(lson, pos, v);
        else update(rson, pos, v);
        pushUp(rt);
    }
}
int query(base, int _l, int _r) {
    if (_l <= l&&_r >= r) return t[rt];
    mid;
    if (_r <= m) return query(lson, _l, _r);
    else if (_l > m) return query(rson, _l, _r);
    else {
        int t1 = query(lson, _l, m);
        int t2 = query(rson, m + 1, _r);
        return max(t1, t2);
    }
    }
}
void main() {
    int num, m;
    while (s("%d%d", &num, &m) != EOF) {
        build(we_base);
        char cmd;
        int in1, in2;
        while (m--) {
            s("\n%c%d%d", &cmd, &in1, &in2);
            if (cmd == 'U') update(we_base, in1, in2);
            else p("%d\n", query(we_base, in1, in2));
        }
    }
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末冠蒋,一起剝皮案震驚了整個(gè)濱河市羽圃,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌抖剿,老刑警劉巖朽寞,帶你破解...
    沈念sama閱讀 206,378評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異斩郎,居然都是意外死亡脑融,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,356評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門缩宜,熙熙樓的掌柜王于貴愁眉苦臉地迎上來肘迎,“玉大人,你說我怎么就攤上這事锻煌〖瞬迹” “怎么了?”我有些...
    開封第一講書人閱讀 152,702評(píng)論 0 342
  • 文/不壞的土叔 我叫張陵炼幔,是天一觀的道長秋茫。 經(jīng)常有香客問我史简,道長乃秀,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,259評(píng)論 1 279
  • 正文 為了忘掉前任圆兵,我火速辦了婚禮跺讯,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘殉农。我一直安慰自己刀脏,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,263評(píng)論 5 371
  • 文/花漫 我一把揭開白布超凳。 她就那樣靜靜地躺著愈污,像睡著了一般。 火紅的嫁衣襯著肌膚如雪轮傍。 梳的紋絲不亂的頭發(fā)上暂雹,一...
    開封第一講書人閱讀 49,036評(píng)論 1 285
  • 那天,我揣著相機(jī)與錄音创夜,去河邊找鬼杭跪。 笑死,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的涧尿。 我是一名探鬼主播系奉,決...
    沈念sama閱讀 38,349評(píng)論 3 400
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢(mèng)啊……” “哼姑廉!你這毒婦竟也來了缺亮?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 36,979評(píng)論 0 259
  • 序言:老撾萬榮一對(duì)情侶失蹤桥言,失蹤者是張志新(化名)和其女友劉穎瞬内,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體限书,經(jīng)...
    沈念sama閱讀 43,469評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡虫蝶,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,938評(píng)論 2 323
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了倦西。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片能真。...
    茶點(diǎn)故事閱讀 38,059評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖扰柠,靈堂內(nèi)的尸體忽然破棺而出粉铐,到底是詐尸還是另有隱情,我是刑警寧澤卤档,帶...
    沈念sama閱讀 33,703評(píng)論 4 323
  • 正文 年R本政府宣布蝙泼,位于F島的核電站,受9級(jí)特大地震影響劝枣,放射性物質(zhì)發(fā)生泄漏汤踏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,257評(píng)論 3 307
  • 文/蒙蒙 一舔腾、第九天 我趴在偏房一處隱蔽的房頂上張望溪胶。 院中可真熱鬧,春花似錦稳诚、人聲如沸哗脖。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,262評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽才避。三九已至,卻和暖如春氨距,著一層夾襖步出監(jiān)牢的瞬間桑逝,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評(píng)論 1 262
  • 我被黑心中介騙來泰國打工衔蹲, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留肢娘,地道東北人呈础。 一個(gè)月前我還...
    沈念sama閱讀 45,501評(píng)論 2 354
  • 正文 我出身青樓,卻偏偏與公主長得像橱健,于是被迫代替她去往敵國和親而钞。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,792評(píng)論 2 345

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