士兵杀敌（数状数组之插线问点）

最新推荐文章于 2024-03-02 09:44:12 发布

原创最新推荐文章于 2024-03-02 09:44:12 发布 · 597 阅读

0 ·

CC 4.0 BY-SA版权

树状数组&&线段树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个用于快速统计和查询特定个体累计军功的算法。该算法通过处理一系列请战和查询指令，在大规模数据集上高效地更新和检索每位士兵的军功值。采用数状数组实现，能够应对频繁的区间加法操作和单点查询需求。

描述
南将军麾下有百万精兵，现已知共有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

数状数组之插线问点

#include "stdio.h"
#include "string.h"
#define MAX 1000010
int data[MAX];


int lowBit (int num) {
	return num & (-num);
}

void plus (int n,int add) {//前n个数加上add

	while (n > 0) {
		data[n] += add;
		n-= lowBit(n);
	}
}

int getValue (int index,int n) {
	int sum = 0;
	while (index <= n) {
		sum += data[index];
		index += lowBit (index);
	}
	return sum;
}
int main () {
	int command;//表示的是指令的个数
	int n;//表示的是士兵的个数
	int i;
	int add;
	char cmd[10];
	int start,end;
	scanf ("%d%d",&command,&n);
	for (i = 0;i < command;i++) {
		scanf ("%s",cmd);
		if (!strcmp (cmd,"ADD")) {
			scanf ("%d%d%d",&start,&end,&add);
			plus(start - 1,-add);
			plus (end,add);
		}else {
			scanf ("%d",&add);
			printf("%d\n",getValue (add,n));
		}
	}
	return 0;
}