一维二维树状数组模板（进阶版）

已于 2022-05-14 21:36:03 修改
阅读量293 收藏 1
点赞数 1
文章标签： 排序算法 算法
于 2022-05-12 21:05:48 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/eyuhaobanga/article/details/124739877
版权

继一维二维树状数组模板题加题解_eyuhaobanga的博客-CSDN博客后，发现树状数组写的太丑了，决定改变一下

一维：

树状数组 1 ：单点修改，区间查询 - LibreOJ 130 - Virtual Judge

AC代码：

#include <bits/stdc++.h>
using namespace std;
// typedef long long i64;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n;
    vector<T> a;
    Fenwick(int n) : n(n), a(n + 1) {}
    void add(int x, T v) {
        for (int i = x; i <= n; i += i & -i) {
            a[i] += v;
        }
    }
    T sum(int x) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            ans += a[i];
        }
        return ans;
    }
    T rangeSum(int l, int r) {
        return sum(r) - sum(l - 1);
    }
};
int n, q;

void Solve() {
    cin >> n >> q;
    Fenwick<i64> fen(n);
    for (int i = 1; i <= n; i++) {
        i64 x;
        cin >> x;
        fen.add(i, x);
    }
    while (q--) {
        int op;
        cin >> op;
        if (op == 1) {
            int i;
            i64 x;
            cin >> i >> x;
            fen.add(i, x);
        }
        else {
            int l, r;
            cin >> l >> r;
            cout << fen.rangeSum(l, r) << '\n';
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int T = 1;
    // cin >> T;
    while (T--) {
        Solve();
    }

    return 0;
}

树状数组 2 ：区间修改，单点查询 - LibreOJ 131 - Virtual Judge

AC代码：

#include <bits/stdc++.h>
using namespace std;
// typedef long long i64;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n;
    vector<T> a;
    Fenwick(int n) : n(n), a(n + 1) {}
    void add(int x, T v) {
        for (int i = x; i <= n; i += i & -i) {
            a[i] += v;
        }
    }
    T sum(int x) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            ans += a[i];
        }
        return ans;
    }
    T rangeSum(int l, int r) {
        return sum(r) - sum(l - 1);
    }
    void rangeAdd(int l, int r, T x) {
        add(l, x);
        add(r + 1, -x);
    }
};


int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, q;
    cin >> n >> q;
    vector<i64> a(n + 1);
    vector<i64> c(n + 1);
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        c[i] = a[i] - a[i - 1];
    }
    Fenwick<i64> fen(n);
    for (int i = 1; i <= n; i++) {
        fen.add(i, c[i]);
    }
    while (q--) {
        int op;
        cin >> op;
        if (op == 1) {
            int l, r;
            i64 x;
            cin >> l >> r >> x;
            fen.rangeAdd(l, r, x);
        }
        else {
            int i;
            cin >> i;
            cout << fen.sum(i) << '\n';
        }
    }

    return 0;
}

树状数组 3 ：区间修改，区间查询 - LibreOJ 132 - Virtual Judge

AC代码：

#include <bits/stdc++.h>
using namespace std;
// typedef long long i64;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n;
    vector<T> a;
    Fenwick(int n) : n(n), a(n + 1) {}
    void add(int x, T v) {
        for (int i = x; i <= n; i += i & -i) {
            a[i] += v;
        }
    }
    T sum(int x) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            ans += a[i];
        }
        return ans;
    }
    T rangeSum(int l, int r) {
        return sum(r) - sum(l - 1);
    }
    void rangeAdd(int l, int r, T x) {
        add(l, x);
        add(r + 1, -x);
    }
};


int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, q;
    cin >> n >> q;
    vector<i64> a(n + 1);
    vector<i64> c(n + 1);
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        c[i] = a[i] - a[i - 1];
    }
    vector<Fenwick<i64>> fen(2, Fenwick<i64> (n));
    for (int i = 1; i <= n; i++) {
        fen[0].add(i, c[i]);
        fen[1].add(i, i * c[i]);
    }
    while (q--) {
        int op;
        cin >> op;
        if (op == 1) {
            int l, r;
            i64 x;
            cin >> l >> r >> x;
            fen[0].add(l, x);
            fen[0].add(r + 1, -x);
            fen[1].add(l, l * x);
            fen[1].add(r + 1, -x * (r + 1));
        }
        else {
            int l, r;
            cin >> l >> r;
            cout << (r + 1) * fen[0].sum(r) - fen[1].sum(r) - l * fen[0].sum(l - 1) + fen[1].sum(l - 1) << '\n';
        }
    }

    return 0;
}

二维：

二维树状数组 1：单点修改，区间查询 - LibreOJ 133 - Virtual Judge

AC代码：

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n, m;
    vector<vector<T>> a;
    Fenwick(int n, int m) : n(n), m(m), a(n + 1, vector<T> (m + 1)) {}
    void add(int x, int y, T v) {
        for (int i = x; i <= n; i += i & -i) {
            for (int j = y; j <= m; j += j & -j) {
                a[i][j] += v;
            }
        }
    }
    T sum(int x, int y) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            for (int j = y; j > 0; j -= j & -j) {
                ans += a[i][j];
            }
        }
        return ans;
    }
    T rangeSum(int a, int b, int c, int d) {
        return sum(a - 1, b - 1) + sum(c, d) - sum(c, b - 1) - sum(a - 1, d);
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, m;
    cin >> n >> m;
    int op;
    Fenwick<i64> fen(n, m);
    while (cin >> op) {
        if (op == 1) {
            int x, y;
            i64 k;
            cin >> x >> y >> k;
            fen.add(x, y, k);
        }
        else {
            int a, b, c, d;
            cin >> a >> b >> c >> d;
            cout << fen.rangeSum(a, b, c, d) << '\n';
        }
    }


    return 0;
}

二维树状数组 2：区间修改，单点查询 - LibreOJ 134 - Virtual Judge

AC代码：

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n, m;
    vector<vector<T>> a;
    Fenwick(int n, int m) : n(n), m(m), a(n + 1, vector<T> (m + 1)) {}
    void add(int x, int y, T v) {
        for (int i = x; i <= n; i += i & -i) {
            for (int j = y; j <= m; j += j & -j) {
                a[i][j] += v;
            }
        }
    }
    T sum(int x, int y) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            for (int j = y; j > 0; j -= j & -j) {
                ans += a[i][j];
            }
        }
        return ans;
    }
    T rangeSum(int a, int b, int c, int d) {
        return sum(a - 1, b - 1) + sum(c, d) - sum(c, b - 1) - sum(a - 1, d);
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, m;
    cin >> n >> m;
    int op;
    Fenwick<i64> fen(n, m);
    while (cin >> op) {
        if (op == 1) {
            int a, b, c, d;
            i64 k;
            cin >> a >> b >> c >> d >> k;
            fen.add(a, b, k);
            fen.add(a, d + 1, -k);
            fen.add(c + 1, b, -k);
            fen.add(c + 1, d + 1, k);
        }
        else {
            int x, y;
            cin >> x >> y;
            cout << fen.sum(x, y) << '\n';
        }
    }


    return 0;
}

二维树状数组 3：区间修改，区间查询 - LibreOJ 135 - Virtual Judge

这个就不放AC代码了，应该是输入有点小问题，所以总是wa，但是结果肯定是对的（应该可以保证）

代码：

#include <bits/stdc++.h>
using namespace std;
using i64 = long long;

template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

template <typename T>
struct Fenwick {
    const int n, m;
    vector<vector<T>> a;
    Fenwick(int n, int m) : n(n), m(m), a(n + 1, vector<T> (m + 1)) {}
    void add(int x, int y, T v) {
        for (int i = x; i <= n; i += i & -i) {
            for (int j = y; j <= m; j += j & -j) {
                a[i][j] += v;
            }
        }
    }
    T sum(int x, int y) {
        T ans = 0;
        for (int i = x; i > 0; i -= i & -i) {
            for (int j = y; j > 0; j -= j & -j) {
                ans += a[i][j];
            }
        }
        return ans;
    }
    T rangeSum(int a, int b, int c, int d) {
        return sum(a - 1, b - 1) + sum(c, d) - sum(c, b - 1) - sum(a - 1, d);
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n, m;
    cin >> n >> m;
    vector<Fenwick<i64>> fen(4, Fenwick<i64> (n, m));
    int op;
    while (cin >> op) {
        if (op == 1) {
            int a, b, c, d;
            i64 x;
            cin >> a >> b >> c >> d >> x;
            fen[0].add(a, b, x), fen[0].add(a, d + 1, -x), fen[0].add(c + 1, b, -x), fen[0].add(c + 1, d + 1, x);
            fen[1].add(a, b, a * x), fen[1].add(a, d + 1, -a * x), fen[1].add(c + 1, b, -a * x), fen[1].add(c + 1, d + 1, a * x);
            fen[2].add(a, b, b * x), fen[2].add(a, d + 1, -b * x), fen[2].add(c + 1, b, -b * x), fen[2].add(c + 1, d + 1, b * x);
            fen[3].add(a, b, a * b * x), fen[3].add(a, d + 1, -a * b * x), fen[3].add(c + 1, b, -a * b * x), fen[3].add(c + 1, d + 1, a * b * x);
        }
        else if (op == 2) {
            int a, b, c, d;
            cin >> a >> b >> c >> d;
            i64 sum0 = a * b * fen[0].sum(a - 1, b - 1) + (c + 1) * (d + 1) * fen[0].sum(c, d) - (c + 1) * b * fen[0].sum(c, b - 1) - a * (d + 1) * fen[0].sum(a - 1, d);
            i64 sum1 = -fen[1].sum(a - 1, b - 1) * (b) - fen[1].sum(c, d) * (d + 1) + fen[1].sum(a - 1, d) * (d + 1) + fen[1].sum(c, b - 1) * b;
            i64 sum2 = -fen[2].sum(a - 1, b - 1) * (a) - fen[2].sum(c, d) * (c + 1) + fen[2].sum(a - 1, d) * (a) + fen[2].sum(c, b - 1) * (c + 1);
            i64 sum3 = fen[3].sum(a - 1, b - 1) + fen[3].sum(c, d) - fen[3].sum(a - 1, d) - fen[3].sum(c, b - 1);
            cout << sum0 + sum1 + sum2 + sum3 << '\n';
        }
    }


    return 0;
}

eyuhaobanga
关注
  • 1
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二维树状数组（模板，单点更新区间求和）
HumveeA6的博客
06-03 355
updata:在（x，y）处加上v query：区间（0，0）到（x，y）的所有数之和 sum：容斥原理求区间所有数之和，(x,y)左上角，(xx,yy)右下角。坐标是从左往右，从上往下增大。 int N; int c[maxn][maxn]; inline int lowbit(int t) { return t&amp;(-t); } void update(int x,int ...
二维树状数组ppt
06-18
二维树状数组模板，，，有一个数列{an}，给出一种操作序列，每次可以改变数列中的一个元素的值 要求动态维护，使得任何时刻都能用较快速度求出数列的部分和
POJ2155 - Matrix （二维树状数组）
weixin_38166726的博客
04-09 208
题目链接： http://poj.org/problem?id=2155 题目大意： 给定一个矩阵，初始化为0，现在可以进行两种操作，一种是查询某个点的值是 0 还是 1。另一种是让这个矩阵的一个子矩阵内的值取反。 解题过程： 省赛选拔赛的题，太难了直接没看……… 后来补起来，有模板还是挺容易的。 题目分析： 首先这题虽然看起来像...
【模板篇】树状数组们（二）
weixin_30265103的博客
02-10 61
上一次，我们说到了，树状数组是可以在O(logn)时间复杂度内，支持单点修改、区间求和的数据结构，还具有 常数小、代码短 的优势。 你忘了？？？ 好吧，把你传送回去看看：http://blog.csdn.net/enzymii/article/details/54952957 那么为什么时间复杂度是O(logn)呢？？？ 下面来证明一下。。。（其实没必要看的2333） 1）首先...
【模板】二维树状数组
zhaoyh2000的博客
02-04 287
相对来说树状数组比线段树扩展到二维好些多了，十几行代码也就。
jsp二维菜单，很好用的东西
03-18
1. **数据结构**：为了存储和处理菜单的层级关系，我们可能需要使用数组、链表或者更复杂的树形数据结构，如ArrayList、LinkedList或自定义的TreeNode类。菜单项通常包含ID、文本、链接和子菜单等属性。 2. **动态...
优雅的将ElementUI表格变身成树形表格的方法步骤
10-17
`arrayToTree`将一维数组转换为树形结构，`ergodicTree`用于遍历树形结构并生成我们需要的表格数据。这两个函数在`./utils/array.js`和`./utils/tree.js`中定义，你需要根据自己的需求来实现它们。 在Vue组件中，...
php实现smarty模板无限极分类的方法
10-23
查询结果存储在一个二维数组`$class_arr`中，其中包含品牌ID、品牌名称、父品牌ID和排序顺序。 接下来，定义了一个名为`dafenglei_select`的递归函数，用于生成HTML的下拉选项。这个函数接受四个参数：当前层级$m、...
kuangbin acm模板超级好用
09-05
3.2.1 一维 . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 57 3.2.2 二维 . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 57 3.3 树链剖分 . . . ....
二维树状数组-模板
amf12345的博客
08-04 301
修改操作如下： void change(int i, int j, int delta){ A[i][j] += delta; for(int x = i; x < A.length; x += lowbit(x)) for(int y = j; y < A[i].length; y += lowbit(y)) C...
二维树状数组模板
我们仍未知道那天所看见的花的名字。
09-05 90
#include <bits/stdc++.h> using namespace std; const int N=5005; int n,m; struct BIT { long long S[N][N]; #define lowbit(x) ((x)&(-x)) inline void Ins(int x,int y,int Tag){ int ly=y; while(x<=n){ while(y.
二维树状数组模板（单点更新，区间求和）（以HDU 2642为例）
常胜将军的博客
10-30 339
题目：点击打开链接 题意：输入B后输入坐标，表示对应的点的灯变亮，输入D后输入坐标表示对应的点灯灭，输入Q后输入一个矩形的左下角和右上角 输出矩形内亮着的等的个数，注意灯亮过不能再亮，灯关了不能再关，所以用数组标记，树状数组模板中元素下标均从1开始，题目从0开始所以加1。 分析：二维树状数组单点更新，区间求和裸题。总结下板子。tree[x][y]记录的是右下角为(x, y)，高为lowbit(x...
See you~ （hdu1892 二维树状数组模板）
残阳止水
07-14 387
See you~
树状数组＆二维树状数组模板
WilliamCode的博客
04-12 774
树状数组模板int lowbit(int x) { return (x&(-x));} int add(int x,int val) { for (;x<=40000;x+=lowbit(x)) c[x]+=val; } int que(int x) { int ans=0; for (;x>0;x-=low
C语言第07章-一维数组和二维数组1完整.pptx
最新发布
11-14
总结来说，一维数组和二维数组是C语言中常用的数据结构，它们可以存储一组具有相同数据类型的数据，并且可以通过下标来访问和操作数组中的元素。一维数组是线性结构，而二维数组是矩阵结构，它由多个一维数组组成。...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值