[模板] 树状数组(单点与区间修改)

.mortal.

已于 2023-04-07 21:56:38 修改

阅读量57

点赞数

分类专栏：数据结构普及模板文章标签： c++ 算法

于 2023-04-07 21:50:55 首次发布

本文链接：https://blog.csdn.net/alone_frank/article/details/130021342

版权

数据结构同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

普及模板

5 篇文章 0 订阅

订阅专栏

文章提供了两种C++代码实现，分别是单点链接和区间链接的BIT（BinaryIndexedTree，二进制索引树）数据结构。BIT用于高效地处理动态数组的区间查询和更新问题，如在常数时间内完成单点修改和区间求和。代码中展示了如何利用低字节运算进行索引计算，并通过差分数组优化区间查询。

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

单点链接

#include <iostream>
#define endl "\n"

const int N = 500010;

using namespace std;

int n, m;

int a[N], c[N];

int lowbit (int x)

{
	return (-x) & x;	
}

void add (int k, int v)

{
	while (k <= n)
	{
		c[k] += v;
		k += lowbit (k);
	}
}

int query (int k)

{
	int s = 0;
	for (int i = k; i > 0; i -= lowbit (i))
	{
		s += c[i];
	}
	return s;
}

int main()

{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	
	cin >> n >> m;
	for (int i = 1; i <= n; i ++) cin >> a[i] , add (i, a[i]);
	while (m --)
	{
		int opt;
		cin >> opt;
		if (opt == 1)
		{
			int k, v;
			cin >> k >> v;
			add (k, v);
		}
		else
		{
			int l, r;
			cin >> l >> r;
			cout << query (r) - query (l - 1) << endl;
		}
	}
}

区间链接

#include <iostream>
#include <cstring>
#include <algorithm>
#define endl "\n"
#define int long long

const int N = 500010;

using namespace std;//BIT

int n, m;

int a[N], tree[N];//原数组 差分数组

int lowbit(int x)
{
    return x & -x;
}

void add (int k, int v)
{
    for (int i = k; i <= n; i += lowbit (i)) tree[i] += v;   
}

int query (int k)

{
    int s = 0;
    for (int i = k; i; i -= lowbit (i)) s += tree[i];
    return s;
}

signed main()

{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    
    
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ ) cin >> a[i];
    for (int i = 1; i <= n; i ++ ) add (i, a[i] - a[i - 1]);//存差分数组
    while (m -- )
    {
        char op;
        cin >> op;
        if (op == '1')
        {
            int l, r, x;
            cin >> l >> r >> x;//构建差分数组 在 a[l] + x , 在a[r + 1] - x;
            add (l, x);
            add (r + 1, -x);
        }
        else 
        {
            int x;
            cin >> x;
            cout << query (x) << endl;//发现求 x 处的值其实就是将差分数组从 tree[1] + tree[2] + ... + tree[k]
        }
    }
}