nyoj 123士兵杀敌（四）

最新推荐文章于 2019-10-22 16:31:14 发布

原创最新推荐文章于 2019-10-22 16:31:14 发布 · 778 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#nyoj123士兵杀敌四 #java #树状数组 #插线问点

算法研究专栏收录该内容

24 篇文章

订阅专栏

本文介绍使用树状数组解决动态更新与查询的问题。通过一个具体的军事背景案例，讲解了如何利用树状数组实现高效的军工数值更新和查询操作。文章详细解释了lowBit(), update() 和 getSum() 方法，并提供了完整的Java代码实现。

题目大意：1~M个士兵，初始的时候，每个人的军工都是零，动态连续增加军工，动态查询某个人的军工！！

思路：由于是动态的，连续插入，单个查询，是典型的插线问点树状数组！！！

对于树状数组的lowBit()、update()和getSum()，多了也就理解了，南阳理工的一些列的《士兵杀敌》做完，理解会更好！

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {
	static int[] arr = new int[1000005];
	static int len;

	public static void main(String[] args) {
		Scanner sc = new Scanner(new BufferedInputStream(System.in));
		int cases;
		String str;

		cases = sc.nextInt();
		len = sc.nextInt();

		while (cases-- > 0) {
			str = sc.next();
			if (str.charAt(0) == 'A') {//看java源码，这个效率高！
				int start = sc.nextInt();
				int end = sc.nextInt();
				int add = sc.nextInt();
				update(start, add);
				update(end + 1, -add);//插线问点，前面增加add,后面要去掉！
			} else {
				System.out.println(getNum(sc.nextInt()));
			}
		}

		sc.close();
	}

	static int lowBit(int x) {
		return -x & x;
	}

	static void update(int x, int add) {
		while (x <= len) {
			arr[x] += add;
			x += lowBit(x);
		}
	}

	static int getNum(int x) {
		int sum = 0;
		while (x > 0) {
			sum += arr[x];
			x -= lowBit(x);
		}
		return sum;
	}
}