博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
数列操作 — 线段树入门
阅读量:3930 次
发布时间:2019-05-23

本文共 1904 字,大约阅读时间需要 6 分钟。

数列操作 — 线段树入门

题目描述 Description

一行N个方格,开始每个格子里都有一个整数。现在动态地提出一些问题和修改:提问的形式是求某一个特定的子区间[a,b]中所有元素的和;修改的规则是指定某一个格子x,加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N<100000,,提问和修改的总数m<10000条。

输入描述 Input Description

输入文件第一行为一个整数N,接下来是n行n个整数,表示格子中原来的整数。接下一个正整数m,再接下来有m行,表示m个询问,第一个整数表示询问代号,询问代号1表示增加,后面的两个数x和A表示给位置X上的数值增加A,询问代号2表示区间求和,后面两个整数表示a和b,表示要求[a,b]之间的区间和。

输出描述 Output Description

共m行,每个整数

样例输入 Sample Input

6
3
4
1 3 5
2 1 4
1 1 9
2 2 6

样例输出 Sample Output

22
22

数据范围及提示 Data Size & Hint

1≤N<100000, m<10000 。

 

思路(见代码注释)

代码

# include 
# define N 500000 // 利用二叉树的父子节点关系来存储,因此必须保证数组开的足够大struct node{ int l,r,v; // left、right、value ;左右孩子由二叉树性质计算。};node st[N];int a[N];void build(int v,int l,int r){ st[v].l = l; st[v].r = r; if (l == r) { // 是叶子,直接赋值,跳出 st[v].v=a[l]; return ; } int mid = (l+r)/2; // 不是叶子,往下扩展 build(2*v,l,mid); build(2*v+1,mid+1,r); st[v].v=st[v*2].v+st[v*2+1].v; // 扩展出孩子后才能计算value域}void insert(int v,int w,int p){ // 在以v为根的树中,寻找节点w,并把节点w的值加上p 。 格外注意 某些编程语言中insert不能作为标识符 if (w>=st[v].l && w<=st[v].r){ st[v].v += p; // 父子之间为统治关系,凡是经过的节点都要更新value } if (st[v].l==st[v].r) return; // 找到终点,跳出 int mid = (st[v].l+st[v].r)/2; if (w<=mid) { // 判断左右孩子 insert(v*2,w,p); }else { insert(v*2+1,w,p); }}int getsum(int v,int l,int r){ // 返回以v为根节点的树中,区间 [l,r] 的value域 if (st[v].l==l && st[v].r==r){ return st[v].v; // 找到,返回 } int mid = (st[v].l+st[v].r)/2; if (r<=mid) { return getsum(v*2,l,r); // 区间 [l,r] 在v的左枝上 }else if (l>mid) { return getsum(v*2+1,l,r); // 右枝 }else { // 兵分两路 return getsum(v*2,l,mid) + getsum(v*2+1,mid+1,r); }}int main(void){ freopen("1080.in","r",stdin); int n; scanf("%d",&n); for (int i(1);i<=n;i++) { scanf("%d",&a[i]); } build(1,1,n); int m,t,a,b; for (scanf("%d",&m);m;m--){ scanf("%d%d%d",&t,&a,&b); if (t & 1) { insert(1,a,b); }else { printf("%d\n",getsum(1,a,b)); } } return 0;}

 

转载地址:http://votgn.baihongyu.com/

你可能感兴趣的文章
项目经理需要具备的技能
查看>>
解读PMP考点:计算题
查看>>
解读PMP考点:项目估算成本级别与估算成本方法
查看>>
解读PMP考点:挣值管理(EVM)
查看>>
解读PMP考点:质量管理中规划质量、实施质量保证、实施质量控制的对比
查看>>
解读PMP考点:PMP考试中关于合同计算类型的题目
查看>>
自动化测试工具介绍
查看>>
你知道这些测试工具吗?
查看>>
测试面试150题(1-50)
查看>>
测试面试150题(51-100)
查看>>
测试面试150题(101-150)
查看>>
为什么要引入自动化测试(Benefits of Test Automation)?
查看>>
基于风险的测试方法(RBT)
查看>>
谈谈 Negative Testing
查看>>
谈谈什么是TDD?
查看>>
对自动化测试常见的5个误解
查看>>
OWASP 2017 年发布的 Top 10安全风险
查看>>
关于安全测试面试的30道基础概念题目
查看>>
Mobile App测试时需要考虑的一些因素
查看>>
一张关于Git的历时发展图(值得一看)
查看>>