树状数组(1)

数据结构:树状数组。

 应用:单点修改;区间查询。

前缀和打表后查询效率是O(1),但后续若修改原数组中的值,需要更新前缀和数组,最坏情况是O(n)。

树状数组的支持单点修改和区间查询,时间复杂度都是O(logn)

例题:1264. 动态求连续区间和 - AcWing题库

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

输入格式

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

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

接下来 mm 行,每行包含三个整数 k,a,bk,a,b (k=0k=0,表示求子数列[a,b][a,b]的和;k=1k=1,表示第 aa 个数加 bb)。

数列从 11 开始计数。

输出格式

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

数据范围

1≤n≤1000001≤n≤100000,
1≤m≤1000001≤m≤100000,
1≤a≤b≤n1≤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>
#include<cstdio>
using namespace std;

int n, m, k;
const int N = 1e5+10;
int f[N], tr[N];

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

void add(int a,int b){
    for(int i = a; i <= n; i += lowbit(i)) tr[i] += b;
}

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

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

1265. 数星星 - AcWing题库

天空中有一些星星,这些星星都在不同的位置,每个星星有个坐标。

如果一个星星的左下方(包含正左和正下)有 kk 颗星星,就说这颗星星是 kk 级的。

例如,上图中星星 55 是 33 级的(1,2,41,2,4 在它左下),星星 2,42,4 是 11 级的。

例图中有 11 个 00 级,22 个 11 级,11 个 22 级,11 个 33 级的星星。

给定星星的位置,输出各级星星的数目。

换句话说,给定 NN 个点,定义每个点的等级是在该点左下方(含正左、正下)的点的数目,试统计每个等级有多少个点。

输入格式

第一行一个整数 NN,表示星星的数目;

接下来 NN 行给出每颗星星的坐标,坐标用两个整数 x,yx,y 表示;

不会有星星重叠。星星按 yy 坐标增序给出,yy 坐标相同的按 xx 坐标增序给出。

输出格式

NN 行,每行一个整数,分别是 00 级,11 级,22 级,……,N−1N−1 级的星星的数目。

数据范围

1≤N≤150001≤N≤15000,
0≤x,y≤320000≤x,y≤32000

输入样例:

5
1 1
5 1
7 1
3 3
5 5

输出样例:

1
2
1
1
0

由于本题输入数据很特殊,所以其实等价于求一下,到目前的输入为止,有多少个星星的 x 值小于等于该星星的 x 就可以了,这就代表该星星的等级。

由于该题y不递减的输入特性,导致了y在题目中毫无作用。

所以我们只需要实时更新一下我们的树状数组就可以了。

正如 y 总所说的,能用到数据结构的算法的题目,往往是根据数据结构的特性来决定的。比如这个题目我们为什么要用树状数组来处理?是因为我们这里要运用前缀和,以及更新前缀和,而这恰好符合树状数组的特性,所以我们利用了它。

#include<iostream>
#include<cstdio>
using namespace std;

const int N = 32010;
int tr[N], ans[N];
int n;

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

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

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

int main(){
    scanf("%d",&n);
    int x, y;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&x,&y);
        x++;
        add(x);
        ans[sum(x)]++;
    }
    
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值