溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

C++ set在樹狀數(shù)組實(shí)現(xiàn)中的簡(jiǎn)化

發(fā)布時(shí)間：2024-08-05 17:10:04 來源：億速云 閱讀：81 作者：小樊 欄目：編程語(yǔ)言

在樹狀數(shù)組實(shí)現(xiàn)中，我們通常需要維護(hù)一個(gè)數(shù)組和一個(gè)set來記錄元素的插入和刪除操作。通過使用C++的set數(shù)據(jù)結(jié)構(gòu)，我們可以簡(jiǎn)化這個(gè)過程，只需要維護(hù)一個(gè)set即可完成元素的插入和刪除操作。

下面是一個(gè)使用C++ set實(shí)現(xiàn)的樹狀數(shù)組簡(jiǎn)化代碼示例：

#include <iostream>
#include <set>
using namespace std;

class FenwickTree {
private:
    set<int> tree;
public:
    void update(int idx, int val) {
        while (idx <= tree.size()) {
            tree.insert(idx);
            idx += idx & -idx;
        }
    }

    int query(int idx) {
        int sum = 0;
        auto it = tree.lower_bound(idx);
        while (it != tree.begin()) {
            sum += *it;
            it--;
            idx -= idx & -idx;
        }
        return sum;
    }
};

int main() {
    FenwickTree ft;
    ft.update(1, 1);
    ft.update(2, 2);
    ft.update(3, 3);
    cout << ft.query(3) << endl; // Output: 6
    return 0;
}

在這個(gè)簡(jiǎn)化的實(shí)現(xiàn)中，我們使用一個(gè)set來代替?zhèn)鹘y(tǒng)的數(shù)組，通過set的lower_bound函數(shù)來查找元素的位置。這樣可以簡(jiǎn)化代碼實(shí)現(xiàn)，并且提高了代碼的可讀性。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

c++
AI
助
手