平时在我们解决区间和问题时,我们用的比较多的算法是前缀和。使用前缀和的话在解题过程中能够降低时间复杂度,是一个比较常见的算法。但是,如果数据区间的数字会进行改动,那么我们就会发现,前缀和算法在这里就不好使了,因此,我们今天需要通过介绍一下一个新的数据结构----线段树。
在此说明,由于篇幅以及笔者能力有限,本篇博客将不介绍线段树的基本定义,建议在观看其它线段树基本概念的博客后再通过本博客来进行进一步加深对线段树的理解。
题目情景引入
Acwing 动态求连续区间和
给定 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
不难发现,如果这道题使用前缀和的话,每进行一次数据的修改都需要更新一次前缀和数组,那么代码的时间复杂度就会高起来了,所以我们舍弃前缀和算法,使用线段树解题。
在这里需要说明的是,如果想要熟悉线段树,光看本篇博客是远远不够的,还需要自己多加练习。
那么接下来我们来看看本题的ac代码
注释:a<<1=a2 ; a<<1|1=a2+1 ; a>>1=a/2 ; l + r >> 1=(l+r)/2;
以上写法为位运算知识
#include<iostream>
#include <utility>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define fi first
#define se second
using namespace std;
typedef pair<int, int> PII;
const int N = 100010;
int n, m;
int w[N];//用来记录每一个点的值
struct node//建立线段树数组
{
int l, r;//左右区间
int sum;//总和
}tr[N * 4];
void push_up(int u)//利用左右儿子来求区间和
{
tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;
}
/*下面三个函数的u都是在主使用时输入1,用来进行线段树的向后查找*/
void build(int u, int l, int r)//建树函数,u是当前点编号,l和r代表这个节点代表的区间范围
{
if (l == r)tr[u] = { l,r,w[r] };
else
{
tr[u] = { l,r };
int mid = l + r >> 1;
build(u << 1, l, mid);//递归求左儿子
build(u << 1 | 1, mid + 1, r);//递归求右儿子
push_up(u);//利用左右儿子来计算一下sum
}
}
int query(int u, int l, int r)//查找l~r区域的和的函数
{
if (l <= tr[u].l && r >= tr[u].r)return tr[u].sum;//线段树被切的区域在l和r内了就把这一块全加上
int mid = tr[u].l + tr[u].r >> 1;//取个中点准备二分
int sum = 0;
if (mid >= l)sum += query(u << 1, l, r);//切块判断左儿子区域是否有与l~r重合的区域
if (r >= mid + 1)sum += query(u << 1 | 1, l, r);//判断右儿子
//这里的递归函数会一直切下去,除非线段区域与l~r不重合
return sum;//返回上一层把每一段sum加起来就是要求的区域的和
}
void modify(int u, int x, int v)//把第x个数修改为v
{
if (tr[u].l == tr[u].r)//此时的l和r都为x,代表找到这个叶节点了
{
tr[u].sum += v;//修改,这里的线段长度就是一个单独的点了
}
else
{
int mid = tr[u].l + tr[u].r >> 1;//开始二分
if (x <= mid)//寻找x点是在现在这条线段的左半边还是右半边,在哪边递归切哪边的线段
{
modify(u << 1, x, v);
}
else
{
modify(u << 1 | 1, x, v);
}
push_up(u);//此时左右儿子都已经递归更新完了,来更新一下父亲,一路递归到最长的线段,即所有数据更新完成
}
}
void solve()
{
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
cin >> w[i];
}
build(1, 1, n);
while (m--)
{
int k, a, b;
cin >> k >> a >> b;
if (k!=0)
{
modify(1, a, b);
}
else
{
cout << query(1, a, b) << endl;
}
}
}
int main()
{
solve();
return 0;
}
到这里,这道题目就这么轻松的解决啦(其实也不轻松 )。
作者:Avalon·Demerzel,如果本篇博客有帮到你的话就留下个赞吧。