数据结构——树状数组与线段树

CSDN729180099

于 2021-02-03 11:34:48 发布

阅读量177

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/CSDN729180099/article/details/113444650

版权

数据结构专栏收录该内容

10 篇文章 0 订阅

订阅专栏

树状数组

基本介绍
本质：用一维数组模拟一棵树
用途：在O(logn)时间内对一段序列进行以下两种操作
1）给某个位置上的数加上一个数
2）求某个前缀和
核心方程：tr[x] = (x - lowbit(x)， x]
即tr[x] = 原数组下标为 (x - lowbit(x)， x] 之间的区间和
两个功能对应的代码：

//给某个位置上的数加上一个数c
for(int i = x; i <= N; i += lowbit(i)) tr[i] += c;  //其中N是数组的最大值

//求某个前缀和
int res = 0;
for(int i = x; i; i -= lowbit(i)) res += tr[i];

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

例题: 动态求连续区间和

给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b （k=0，表示求子数列[a,b]的和；k=1，表示第 a 个数加 b）。

数列从 1 开始计数。

输出格式
输出若干行数字，表示 k=0 时，对应的子数列 [a,b] 的连续和。

数据范围
1≤n≤100000,
1≤m≤100000，
1≤a≤b≤n
输入样例：
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例：
11
30
35

#include<iostream>
#include<algorithm>
#include<cstdio>

using namespace std;

const int N = 100010;

int n, m;
int tr[N];
int a[N];

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

void add(int x, int c)
{
    for(int i = x; i < N; i += lowbit(i)) tr[i] += c;
}

int query(int x)
{
    int res = 0;
    for(int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
    for(int i = 1; i <= n; i ++) add(i, a[i]); 
    
    while(m--)
    {
        int f, a, b;
        cin >> f >> a >> b;
        if(f)
            add(a, b);
        else 
            printf("%d\n", query(b) - query(a-1));
    }
    
    return 0;
}

线段树

基本介绍
本质：利用二叉树维护一个一维数组
用途：在O(logn)时间内对一段序列进行以下两种操作
1）修改某个位置上的数
2）求某个区间和等
基本操作对应的代码：

//用节点u的两个孩子节点更新u
void pushup(int u)
{
    tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;
}

//建立以u为根节点，包含区间为(l, r)的树
void build(int u, int l, int r)
{
    if(l == r) tr[u] = {l, r, a[l]};    // 当区间长度为1时，为叶子节点，直接建立
    else                                // 当区间长度不为1时，递归建立左右子树
    {
        tr[u] = {l, r};
        int mid = l + r >> 1;
        build(u << 1, l, mid);
        build(u << 1 | 1, mid + 1, r);
        pushup(u);
    }
}

//修改下标为x的节点，给下标为x的节点+c
void modify(int u, int x, int c)
{
    if(tr[u].l == tr[u].r) tr[u].sum += c;  //叶子节点直接加
    else                                    //跟新包含该下标的节点
    {
        int mid = tr[u].l + tr[u].r >> 1;
        if(mid >= x) modify(u << 1, x, c);
        else modify(u << 1 | 1, x, c);
        pushup(u);
    }
}

//查询区间(l, r)的和
int query(int u, int l, int r)
{
    if(l <= tr[u].l && r >= tr[u].r) return tr[u].sum;  //如果区间完全被包含，直接返回和值
    //递归寻找到完全被包含区间节点
    int mid = tr[u].l + tr[u].r >> 1;                   
    int sum = 0;
    if(l <= mid) sum += query(u << 1, l, r);
    if(r > mid) sum += query(u << 1 | 1, l, r);
    
    return sum;
}

例题：动态求连续区间和