实用数据结构---树状数组(二叉索引树)

树状数组适用于动态连续和查询问题,就是给定一个区间,

查询某一段的和或者修改某一位置的值。

关于树状数组的结构请去百度百科,否则将看不懂下面内容

我们看这个题

士兵杀敌(二)

时间限制:1000 ms  |  内存限制:65535 KB 难度:5

描述

南将军手下有N个士兵,分别编号1N,这些士兵的杀敌数都是已知的。

小工是南将军手下的军师,南将军经常想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。

南将军的某次询问之后士兵i可能又杀敌q人,之后南将军再询问的时候,需要考虑到新增的杀敌数。

输入

只有一组测试数据
第一行是两个整数N,M,其中N表示士兵的个数(1<N<1000000)M表示指令的条数。(1<M<100000)
随后的一行是N个整数,ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行是一条指令,这条指令包含了一个字符串和两个整数,首先是一个字符串,如果是字符串QUERY则表示南将军进行了查询操作,后面的两个整数m,n,表示查询的起始与终止士兵编号;如果是字符串ADD则后面跟的两个整数I,A(1<=I<=N,1<=A<=100),表示第I个士兵新增杀敌数为A.

输出

对于每次查询,输出一个整数R表示第m号士兵到第n号士兵的总杀敌数,每组输出占一行

样例输入

5 6

1 2 3 4 5

QUERY 1 3

ADD 1 2

QUERY 1 3

ADD 2 3

QUERY 1 2

QUERY 1 5

样例输出

6

8

8

20

这就是典型的动态区间连续和查询问题。

树状数组需要三个函数

int lowBit(int x){
	return x&(-x);	//返回的值是2的k次幂,k为x二进制下从低位到高位0的个数
}
int sum(int n){
	int res = 0;
	while (n > 0){
		res += c[n];
		n -= lowBit(n);
	}
	return res;
}
void change(int i, int x, int n){
	if (i > n) return;
	c[i] += x;
	change(i + lowBit(i), x, n);
}

对于lowBit我解释一下,为什么是 x & -x,计算机里的整数采用补码表示
因此-x其实是按位取反,然后在末尾加1的结果,举个例子
32388 = 1001010110010000
-32388 = 0110101001110000
所以lowBit返回的值是x在二进制下从低位到高位最右面的1所对应的值
完整程序在下面

#include<stdio.h>
#define MAX_NUM 1000005
int a[MAX_NUM] = { 0 };
int c[MAX_NUM] = { 0 };
int lowBit(int x){
	return x&(-x);	//返回的值是2的k次幂,k为x二进制下从低位到高位0的个数
}
int sum(int n){
	int res = 0;
	while (n > 0){
		res += c[n];
		n -= lowBit(n);
	}
	return res;
}
void change(int i, int x, int n){
	if (i > n) return;
	c[i] += x;
	change(i + lowBit(i), x, n);
}
int main(){
	int n, m;
	scanf("%d%d", &n, &m);
	int i;
	for (i = 1; i <= n; i++){
		scanf("%d", &a[i]);
		if (i % 2 != 0)
			c[i] = a[i];
		else{
			int j;
			for (j = i - lowBit(i) + 1; j <= i; j++)
				c[i] += a[j];
		}
	}
	char order[10];
	int b, e, t = 0;
	for (i = 0; i < m; i++){
		scanf("%s", &order);
		scanf("%d%d", &b, &e);
		if (order[0] == 'Q'){
			printf("%d\n", sum(e) - sum(b - 1));
		}
		else{
			change(b, e, n);
		}
	}
	return 0;
}

推荐一个公众号,不吐槽,不毒舌,偶尔发发文章,偶尔推荐好物,欢迎关注或者有女票的程序狗们推荐给女票

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值