【每日基础算法】树状数组 - 动态求连续区间和

举报
苏州程序大白 发表于 2022/04/21 12:01:14 2022/04/21
【摘要】 @[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]的连续和。

输入格式

第一行包含两个整数 nm,分别表示数的个数和操作次数。

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

接下来 m 行,每行包含三个整数k, a, bk = 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

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。