首先线上模板例题,来自洛谷。
题目描述
如题,已知一个数列,你需要进行下面两种操作:
-
将某一个数加上 xx
-
求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,mn,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nn 个用空格分隔的整数,其中第 ii 个数字表示数列第 ii 项的初始值。
接下来 mm 行每行包含 33 个整数,表示一个操作,具体如下:
-
1 x k
含义:将第 xx 个数加上 kk -
2 x y
含义:输出区间 [x,y][x,y] 内每个数的和
输出格式
输出包含若干行整数,即为所有操作 22 的结果。
输入输出样例
输入 #1复制
5 5 1 5 4 2 3 1 1 3 2 2 5 1 3 -1 1 4 2 2 1 4
输出 #1复制
14 16
说明/提示
【数据范围】
对于 30\%30% 的数据,1 \le n \le 81≤n≤8,1\le m \le 101≤m≤10;
对于 70\%70% 的数据,1\le n,m \le 10^41≤n,m≤104;
对于 100\%100% 的数据,1\le n,m \le 5\times 10^51≤n,m≤5×105。
样例说明:
故输出结果14、16
线段数其实非常方便理解,可以理解成是二分搜索,加上递归排序的思想。
下面附上这道题的代码和注释。
#include<iostream>
using namespace std;
int f[2000010],a[500010];
void buildtree(int k, int l, int r)//建树操作(k是下标,l是区间左端,r是区间右端)
{
if (l == r)//l等于r是说明l到达了区间最右端。建树准备结束。
{
f[k] = a[l];//那么下标k就直接确定为a[l]。
return;//返回。
}
int m = (l + r) >> 1;//开始递归
buildtree(k + k, l, m);
buildtree(k + k + 1, m + 1, r);
f[k] = f[k + k] + f[k + k + 1];//f[k]表示的是这一区间的和,也就是以k为父节点,2k和2k+1为子节点相加的和。
}
void add(int k, int l, int r, int x, int y)//添加操作
{
f[k] += y;//加上y。
if (l == r)
return;
int m = (l + r) >> 1;
if (x <= m)//如果x在m左边
{
add(k + k, l, m, x, y);//在左儿子处递归
}
else add(k + k + 1, m + 1, r, x, y);//x在m右边,在右儿子处递归。
}
int yuuka(int k, int l, int r, int s, int t)//区间求和,s和t分别是目标区间的左右端点。
{
if (l == s && r == t)
return f[k];//找到吻合的区间输出f[k]
int m = (r + l) >> 1;
if (t <= m)//如果目标区间全在左侧
{
return yuuka(k + k, l, m, s, t);
}
else
if (s > m)//目标区间全在右侧
return yuuka(k + k + 1, m + 1, r, s, t);
else//目标区间左右都有
{
return yuuka(k + k, l, m, s, m) + yuuka(k + k + 1, m + 1, r, m+1, t);
}
}
int main()
{
int n,m;
cin >> n>>m;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
buildtree(1, 1, n);
for (int i = 0; i < m; i++)
{
int t, x, y;
cin >> t >> x >> y;
if (t == 1)
{
add(1, 1, n, x, y);
}
else
cout << yuuka(1, 1, n, x, y)<<endl;
}
}