区间 [1,4) 的和

👁️ 0 人浏览 💬 0 人评论 ❤️ 添加收藏

运行下面这段程序:

本节模型:迭代线段树 seg_build(a) 返回 Seg{n, tree},seg_query(s, l, r) 求 a[l..r-1](左闭右开)的和。

#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
using namespace std;

struct Seg {
    int n;
    vector<long long> tree;
};

Seg seg_build(const vector<int>& a) {
    // 迭代线段树(区间和):叶子在 tree[n..2n-1],父亲 i 的孩子是 2i、2i+1
    int n = (int)a.size();
    vector<long long> tree(2 * n, 0);
    for (int i = 0; i < n; i++) {
        tree[n + i] = a[i];
    }
    for (int i = n - 1; i > 0; i--) {
        tree[i] = tree[2 * i] + tree[2 * i + 1];
    }
    return {n, tree};
}

long long seg_query(const Seg& s, int l, int r) {
    // 区间和 a[l..r-1](左闭右开)
    long long res = 0;
    l += s.n;
    r += s.n;
    while (l < r) {
        if (l & 1) {
            res += s.tree[l];
            l++;
        }
        if (r & 1) {
            r--;
            res += s.tree[r];
        }
        l /= 2;
        r /= 2;
    }
    return res;
}

int main() {
    cout << seg_query(seg_build({1, 2, 3, 4, 5}), 1, 4) << endl;
}

(本题用 g++ -std=c++17 -O0 编译。)

提交你的答案
请登录后提交答案。
去登录
代码编辑器
Ctrl + Enter 运行
本次输入:
输出:

                        
👩‍🏫
AI
💬 题目评论

全部评论