【每日基础算法】树状数组 - 动态求连续区间和
【摘要】 @[TOC](【每日基础算法】树状数组 - 动态求连续区间和) 博主介绍🌊 作者主页:苏州程序大白🌊 作者简介:🏆CSDN人工智能域优质创作者🥇,苏州市凯捷智能科技有限公司创始之一,目前合作公司富士康、歌尔等几家新能源公司💬如果文章对你有帮助,欢迎关注、点赞、收藏(一键三连)和C#、Halcon、python+opencv、VUE、各大公司面试等一些订阅专栏哦💅 有任何问题欢迎私...
@[TOC](【每日基础算法】树状数组 - 动态求连续区间和)
博主介绍
🌊 作者主页:苏州程序大白
🌊 作者简介:🏆CSDN人工智能域优质创作者🥇,苏州市凯捷智能科技有限公司创始之一,目前合作公司富士康、歌尔等几家新能源公司
💬如果文章对你有帮助,欢迎关注、点赞、收藏(一键三连)和C#、Halcon、python+opencv、VUE、各大公司面试等一些订阅专栏哦
💅 有任何问题欢迎私信,看到会及时回复
💅关注苏州程序大白,分享粉丝福利
功能
-
让某个位置上的数加上一个数
-
求某一个前缀和
操作
-
lowbit(x):返回x的最后一位1
-
add(x,v):在x位置加上v,并将后面相关联的位置也加上v
-
query(x):询问x的前缀和
-
c[x]:表示的区间和是(x−lowbit(x),x]
add(x,k)操作
需要让后面所有包含元素区间和都增加K
for (int i = x; i <= n; i += lowbit(i)) {
c[i] += k;
}
query(x)操作
需要累加X前面全部的元素,每个包含了i - lowbit(i))
的数
for (int i = x; i; i -= lowbit(i)) {
sum += c[i];
}
案例:动态求连续区间和
给定 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,
数据保证在任何时候,数列中所有元素之和均在 int 范围内。
输入样例:
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>
using namespace std;
const int N = 1e5 + 10;
int n, m;
int c[N];
//lowbit返回二进制第一个非零的1
int lowbit(int x) {
return x & -x;
}
//在第x个位置上,加上k
void add(int x, int k) {
for (int i = x; i <= n; i += lowbit(i)) c[i] += k;
}
//返回x的前缀和
int query(int x) {
int res = 0;
for (int i = x; i; i -= lowbit(i)) res += c[i];
return res;
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) {
int x;
scanf("%d", &x);
add(i, x);
}
while (m--) {
int k, a, b;
scanf("%d %d %d", &k ,&a, &b);
if (k) add(a, b);
else printf("%d\n", query(b) - query(a - 1));
}
return 0;
}
【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
评论(0)