首页 > 代码库 > nyoj 123 士兵杀敌(四)【树状数组】+【插线问点】

nyoj 123 士兵杀敌(四)【树状数组】+【插线问点】

树状数组有两种情况:插点问线和插线问点。这道题是插线问点。

由于树状数组最简单的作用是计算1~x的和,所以给出(a, b。 c)。表示(a,b)区间添加c, 那我们仅仅须要在a点原来的基础上添加c,然后在b点原来的基础上更新-c,这样我们算终于结果的时候在(a。 b)之间的就是添加了c。在区间之外的就是没有添加。

代码:

#include <stdio.h>
#include <string.h>
#define M 1000005
int c[M], m;

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

int getsum(int x){
	int sum = 0;
	while(x){
		sum += c[x];
		x -= lowbit(x);
	}
	return sum;
}

void add(int x, int val){
	while(x <= m){
		c[x] += val;
		x += lowbit(x);
	}
}

int main(){
	int t;
	scanf("%d%d", &t, &m);
	memset(c, 0, sizeof(c));
	char s[9];
	int a, b, c;
	while(t --){
		scanf("%s", s);
		if(s[0] == ‘A‘){
			scanf("%d%d%d", &a, &b, &c);
			add(a, c);
			add(b+1, -c);
		}
		else{
			scanf("%d", &a);
			printf("%d\n", getsum(a));
		}
	}
}  

题目链接:http://acm.nyist.net/JudgeOnline/problem.php?pid=123


nyoj 123 士兵杀敌(四)【树状数组】+【插线问点】