nyoj108 士兵杀敌(一) (线段树)
2016-02-28 20:19
393 查看
士兵杀敌(一)
描述南将军手下有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
线段树初学者,区间求和。
线段树学习,推荐:/article/8334307.html
#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; }
相关文章推荐
- 2015年终总结
- 数据结构-7
- Android 4.2蓝牙介绍
- 关于github中的README.md文件
- SCU 1114-数字三角(dp)
- swift之NSString完成字符串的查找
- android跨进程通信(IPC):使用AIDL
- 挖坑
- 使用lnmp一键安装包后yum源出现的问题与解决
- hdoj Jam's math problem 5615 (数学十字相乘法)暴力
- 无法打开lib文件的解决方法
- 抽象类和接口
- java语言基础,学习笔记
- Hadoop 1.x 启动脚本分析
- MapReduce的写法
- POJ1154(DFS回溯)
- css3动画
- django-admin.py在win7 64位下不能创建工程、项目和执行命令的解决办法
- margin
- 集合Set