nyoj108 士兵杀敌（一） （线段树）

士兵杀敌（一）

描述

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

小工是南将军手下的军师，南将军现在想知道第m号到第n号士兵的总杀敌数，请你帮助小工来回答南将军吧。

注意，南将军可能会问很多次问题。

输入

只有一组测试数据
第一行是两个整数N,M，其中N表示士兵的个数(1<N<1000000)，M表示南将军询问的次数(1<M<100000)
随后的一行是N个整数，ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行有两个整数m,n，表示南将军想知道第m号到第n号士兵的总杀敌数（1<=m,n<=N)。

输出

对于每一个询问，输出总杀敌数
每个输出占一行

样例输入

5 2
1 2 3 4 5
1 3
2 4

样例输出

6
9

线段树初学者，区间求和。

线段树学习，推荐：http://blog.csdn.net/metalseed/article/details/8039326

#include <iostream>
#include<algorithm>
#include <cstdio>
#include <cstring>
#define MAX 1000010
using namespace std;
int array[MAX];
int segTree[MAX * 4 + 10];		//线段树数组大小约为原数组大小的4倍 
void build(int node, int begin, int end) {
	if(begin == end) {
		segTree[node] = array[begin];		//叶节点 
	}
	else {
		build(node * 2, begin, (begin + end) / 2);
		build(node * 2 + 1, (begin + end) / 2 + 1, end);
		segTree[node] = segTree[node * 2] + segTree[node * 2 + 1];		//对两个孩子求和 
	}
}
int query(int node, int begin, int end, int left, int right) {
	if(begin == left && end == right) {
		return segTree[node];
	}
	else {		//将区间一分为二，讨论三种情况 
		int mid = (begin + end) / 2;
		if(right <= mid) return query(node * 2, begin, mid, left, right);
		else if(left > mid) return query(node * 2 + 1, mid + 1, end, left, right);
		else return query(node * 2, begin, mid, left, mid) + query(node * 2 + 1, mid + 1, end, mid + 1, right);  //查询区间和原始区间都要对半分。
	}
}
int main() {
    int N, M;
    scanf("%d %d", &N, &M);
    int i;
    for(i = 1; i <= N; i++) {
    	scanf("%d", &array[i]);
	}
	build(1, 1, N);
	int left, right;
	while(M--) {
		scanf("%d %d", &left, &right);
		printf("%d\n", query(1, 1, N, left, right));
	}
    return 0;
}