NYOJ 题目116士兵杀敌（二）

士兵杀敌（二）

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

难度： 5

描述

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

小工是南将军手下的军师，南将军经常想知道第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

             线段树的区间查询及线段数的更新

       

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

public class Main {

	public static void main(String[] args) {
		Scanner cin = new Scanner(new BufferedInputStream(System.in));
		
		int N = cin.nextInt();
		int M = cin.nextInt();
		
		int[] array = new int[N];
		for(int i=0; i<N; i++) {
			array[i] = cin.nextInt();
		}
		
		nodes = new Node[N * 2 - 1];
		buildTree(array, 0, N - 1, 0);
		
		for(int i=0; i<M; i++) {
			String cmd = cin.next();
			int a = cin.nextInt();
			int b = cin.nextInt();
			if("QUERY".equals(cmd)) {
				System.out.println(getSum(array, 0, N - 1, a - 1, b - 1, 0));
			} else {
				update(a - 1, 0, b);
			}
		}
		
		cin.close();
	}

	static Node[] nodes;

	static int buildTree(int[] array, int left, int right, int index) {
		if (left == right) {
			if (index >= nodes.length) {
				return 0;
			}
			nodes[index] = new Node(left, right, array[left]);
			return array[left];
		}

		int mid = (left + right) / 2;
		int lsum = buildTree(array, left, mid, index * 2 + 1);
		int rsum = buildTree(array, mid + 1, right, index * 2 + 2);
		nodes[index] = new Node(left, right, lsum + rsum);
		return lsum + rsum;
	}

	static int getSum(int[] array, int low, int high, int s, int e, int index) {
		if (s <= low && e >= high) {
			return nodes[index].sum;
		}
		if (s > high || e < low) {
			return 0;
		}
		int mid = (low + high) / 2;
		int lsum = getSum(array, low, mid, s, e, index * 2 + 1);
		int rsum = getSum(array, mid + 1, high, s, e, index * 2 + 2);
		return lsum + rsum;
	}

	static void update(int arrayIndex, int treeIndex, int value) {
		Node node = nodes[treeIndex];
		if (node.left == node.right && node.left == arrayIndex) {
			node.sum += value;
		} else {
			int mid = (node.left + node.right) / 2;
			if (arrayIndex <= mid) {
				update(arrayIndex, treeIndex * 2 + 1, value);
			} else {
				update(arrayIndex, treeIndex * 2 + 2, value);
			}
			node.sum += value;
		}
	}
}

class Node {
	int left;
	int right;
	int sum;

	Node(int l, int r, int s) {
		this.left = l;
		this.right = r;
		this.sum = s;
	}

	@Override
	public String toString() {
		return "(" + left + "," + right + "," + sum + ")";
	}

}