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 61 2 3 4 5QUERY 1 3ADD 1 2QUERY 1 3ADD 2 3QUERY 1 2QUERY 1 5

样例输出

68820

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

       

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;}@Overridepublic String toString() {return "(" + left + "," + right + "," + sum + ")";}}