士兵杀敌（四）

士兵杀敌（四）

时间限制： 2000 ms  |  内存限制： 65535 KB

难度： 5

描述

南将军麾下有百万精兵，现已知共有M个士兵，编号为1~M，每次有任务的时候，总会有一批编号连在一起人请战（编号相近的人经常在一块，相互之间比较熟悉），最终他们获得的军功，也将会平分到每个人身上，这样，有时候，计算他们中的哪一个人到底有多少军功就是一个比较困难的事情，军师小工的任务就是在南将军询问他某个人的军功的时候，快速的报出此人的军功，请你编写一个程序来帮助小工吧。

假设起始时所有人的军功都是0.

输入

只有一组测试数据。
每一行是两个整数T和M表示共有T条指令，M个士兵。（1<=T,M<=1000000)
随后的T行，每行是一个指令。
指令分为两种：
一种形如
ADD 100 500 55 表示，第100个人到第500个人请战，最终每人平均获得了55军功，每次每人获得的军功数不会超过100，不会低于-100。
第二种形如：
QUERY 300 表示南将军在询问第300个人的军功是多少。

输出

对于每次查询输出此人的军功，每个查询的输出占一行。

样例输入

4 10
ADD 1 3 10
QUERY 3
ADD 2 6 50
QUERY 3

样例输出

10
60

一开始用线段树，超时。后来...

树状数组：插线问点

定义数组：kills[1000001]

kills[i] : 表示编号为： lowbit(i)+1 到 i 每个士兵的杀敌数，一开始是全部为0，因此不用 build()

代码如下：

#include<stdio.h>
#include<string.h>

int kills[1000001];	// kills[i] : 表示编号为： lowbit(i)+1 到 i 每个士兵的杀敌数，一开始是全部为0，因此不用 build()

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

// 等价于为 1--right 号士兵增加杀敌数：num 个，然后 1--left -1 号士兵减少杀敌数 num 个
void add( int left, int right, int num )
{
	if( 1 == left )
	{
		for( int i = right; i > 0; i -= lowbit(i) )
		{
			kills[i] += num;
		}
	}
	else 
	{
		add( 1, left - 1, -num );
		add( 1, right, num );
	}
}
int query( int index )
{
	int total = 0;
	for( int i = index; i < 1000001; i += lowbit(i) )
	{
		total += kills[i];
	}
	return total;
}



int main()
{
	int N,Q;
	char c[8];
	int a,b,n;

	scanf( "%d %d", &Q,&N );
	while( Q-- )
	{
		scanf( "%s", c );
		if( strcmp( c, "ADD" ) == 0 )
		{
			scanf( "%d%d%d", &a, &b, &n );
			add( a, b, n );
		}
		else if( strcmp( c, "QUERY" ) == 0 )
		{
			scanf( "%d", &n );
			printf("%d\n", query( n ) );
		}
	}
//	getchar();
//	getchar();

}