树状数组模版

最新推荐文章于 2024-11-18 18:10:05 发布

yan_qiu_ynlchrz

最新推荐文章于 2024-11-18 18:10:05 发布

阅读量69

点赞数

分类专栏：算法整理文章标签：算法

本文链接：https://blog.csdn.net/yan_qiu_ynlchrz/article/details/115978090

版权

算法整理专栏收录该内容

17 篇文章 0 订阅

订阅专栏

本文介绍了树状数组的基础概念，并详细讲解了一维和二维树状数组的实现与应用场景，帮助读者理解这一高效的数据结构在算法中的运用。

摘要由CSDN通过智能技术生成

一维树状数组

struct Fenwick
{
    LL d[N];

    void init(){ rep(i, n + 1) d[i] = 0; }
    void modify(int x, LL val){ while (x <= n){ d[x] += val; x += x & -x; } }
    LL query(int x){ LL ret = 0; while (x){ ret += d[x]; x -= x & -x; } return ret; }
} fen;

二维树状数组

struct TwoD_Fenwick
{
    LL d[N][N];

    void init()
    {
        rep(i, n + 1) rep(j, m + 1) d[i][j] = 0;
    }
    void add(int x, int y, LL val)
    {
        for (int i = x; i <= n; i += i & -i)
            for (int j = y; j <= m; j += j & -j)
                d[i][j] += val;
    }
    LL query(int x, int y)
    {
        LL ret = 0;
        for (int i = x; i > 0; i -= i & -i)
            for (int j = y; j > 0; j -= j & -j)
                ret += d[i][j];
        return ret;
    }
} fen;