nyoj123 士兵杀敌(四) (线段树)
2016-03-01 21:42
393 查看
士兵杀敌(四)
时间限制:2000 ms | 内存限制:65535 KB难度:5
描述
南将军麾下有百万精兵,现已知共有M个士兵,编号为1~M,每次有任务的时候,总会有一批编号连在一起人请战(编号相近的人经常在一块,相互之间比较熟悉),最终他们获得的军功,也将会平分到每个人身上,这样,有时候,计算他们中的哪一个人到底有多少军功就是一个比较困难的事情,军师小工的任务就是在南将军询问他某个人的军功的时候,快速的报出此人的军功,请你编写一个程序来帮助小工吧。
假设起始时所有人的军功都是0.
输入
只有一组测试数据。
每一行是两个整数T和M表示共有T条指令,M个士兵。(1<=T,M<=1000000)
随后的T行,每行是一个指令。
指令分为两种:
一种形如
ADD 100 500 55 表示,第100个人到第500个人请战,最终每人平均获得了55军功,每次每人获得的军功数不会超过100,不会低于-100。
第二种形如:
QUERY 300 表示南将军在询问第300个人的军功是多少。
输出
对于每次查询输出此人的军功,每个查询的输出占一行。
样例输入
4 10 ADD 1 3 10 QUERY 3 ADD 2 6 50 QUERY 3
样例输出
10 60
本以为是区间更新,但整了好久没写出来,后来看了大神的博客才发现原来更新也能这么巧妙。自己学的不扎实,还太死板。。。
在插入数据的时候,遇到第一个完全包含与要插入的区间时,将数据加到该区间。然后每一次查询,将id通过的每一个区间的数据相加。
参考博客:/article/4726337.html
感谢!!
另外:将区间树每个节点的左右边界保存在结构体内可以节省大量时间。除此之外,cin,cout要少用啊,一不小心就超时了。。。
#include <iostream> #include<algorithm> #include <cstdio> #include <cstring> #define MAXN 1000100 using namespace std; struct node { int l, r; int val; }; node T[MAXN * 4 + 10]; void build(int root, int begin, int end) { T[root].l = begin; T[root].r = end; T[root].val = 0; if(begin < end) { int mid = (begin + end) >> 1; build(root << 1, begin, mid); build(root << 1 | 1, mid + 1, end); } } void updata(int root, int l, int r, int add) { if(T[root].l >= l && T[root].r <= r) { T[root].val += add; } else { int mid = (T[root].l + T[root].r) >> 1; if(r <= mid) updata(root << 1, l, r, add); else if(l > mid) updata(root << 1 | 1, l, r, add); else { updata(root << 1, l, mid, add); updata(root << 1 | 1, mid + 1, r, add); } } } int query(int root, int id) { if(T[root].l == T[root].r) { return T[root].val; } else { int sum = T[root].val; int mid = (T[root].l + T[root].r) >> 1; if(id <= mid) sum += query(root << 1, id); else sum += query(root << 1 | 1, id); return sum; } } int main() { //freopen("input.txt", "r", stdin); int t, M; scanf("%d %d", &t, &M); build(1, 1, M); char str[10]; while(t--) { scanf("%s", str); if(str[0] == 'A') { int l, r, add; scanf("%d %d %d", &l, &r, &add); updata(1, l, r, add); } else { int id; scanf("%d", &id); printf("%d\n", query(1, id)); } } return 0; }
相关文章推荐
- 第一章 Java Web开发简介
- 精美viso制图(1)
- python小白-day7 socket初识
- 江西承受得起爱情和事业及更好的软件团队-2016年2月江西IDC排行榜与发展报告
- MySQL常用的命令添加用户、授权、及删除用户
- Android杂项
- 机房重构---七层登录
- 救济金发放(The Dole Queue, UVa 133)
- IOS-APP提交上架流程(新手必看!2016年3月1日最新版)
- 怎样衡量一个组员在团队中的Performance
- atm32的sram DATA_IN_ExtSRAM
- MyBatis入门1--初识Mybatis
- 深度神经网络结构以及Pre-Training的理解
- 一道面试题
- 在官网下载了最新版的PHP,解压后的安装包里为什么没有php5isapi.dll这个dll文件?
- 二叉树的遍历、求深与还原
- CSS3媒介查询
- WPF 利用Process.Start()方法启动指定路径下的exe文件并传递参数
- OpenCL函数-1
- 14条军规