LeetCode(307) Range Sum Query - Mutable
来源:互联网 发布:淘宝大学商学院杭州 编辑:程序博客网 时间:2024/06/08 17:35
题目
Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.
The update(i, val) function modifies nums by updating the element at index i to val.
Example:
Given nums = [1, 3, 5]
sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8
Note:
The array is only modifiable by the update function.
You may assume the number of calls to update and sumRange function is distributed evenly.
分析
一个可变数组数据结构的实现,包含更新数组元素和求部分和两个函数。
乍一看好像一个很简单的题目,很容易便可以 根据下标O(1)更新元素,由遍历操作O(n)实现求和;但是得到TLE的提交反馈也应该在意料之中。
查阅资料,发现一个树状数组的解法,给出参考博客:网址
树状数组是一种用于通过辅助数组和位置算法实现动态管理部分和的一种算法,其核心函数为lowbit、add、sum,当我们需要加入一个元素时,使用add函数向某个位置注入值,当需要求0~某位置的和时,调用sum函数传入这个位置即可:
int lowbit(int pos){ return pos&(-pos); } void add(int pos, int value){ while(pos < c.size()){ c[pos] += value; pos += lowbit(pos); } } int sum(int pos){ int res = 0; while(pos > 0){ res += c[pos]; pos -= lowbit(pos); } return res; }
其中c就是辅助数组。需要注意的是树状数组的索引必须从1开始,因此在与题目的输入量相转化时,需要把索引+1作为树状数组的索引。
假设我们有数组arr:{1,2,3},要求部分和,我们先把这些值通过add注入树状数组,一定要注意索引+1.
for(int i = 0; i < 3; i++){ add(i+1,arr[i]); // 把arr[i]添加到树状数组的i+1位置}
操作完成后,树状数组就建立好了,下面就可以利用sum来求和了。例如我们需要arr中索引0~2的和,那么使用sum(3)即可,如果要求1~2的和,可以用sum(3)-sum(0),注意到,sum(0)对于树状数组是一个非法索引,但是通过观察sum函数发现pos=0正好返回0,也就是到这个位置的和为0,满足要求,不必特殊处理。
综上所述,要求arr中i~j的部分和,使用sum(j+1)-sum(i)即可。
值的更新问题,题目要求使用update函数更新arr中位置i的值为val,这就要利用add函数来实现,注意add函数是在位置pos上追加一个值value,而不是覆盖,因此我们需要计算值的变化量,把它追加到相应位置,并且一定要记得更新arr,否则下次得到的变化量是错误的。
void update(int i, int val) { int ori = m_nums[i]; // m_nums是拷贝arr数组所得的成员变量 int delta = val - ori; m_nums[i] = val; add(i+1,delta);}
所谓树状数组也是第一次了解到,还需查阅资料深入学习一下。
AC代码
//普通方法实现class NumArray1 {public: NumArray1(vector<int> &nums) { array = vector<int>(nums.begin(), nums.end()); int len = array.size(), tmpSum = 0; for (int i = 0; i < len; ++i) { tmpSum += array[i]; allSum.push_back(tmpSum); }//for } void update(int i, int val) { if (i < 0 || i >= array.size()) return; int tmp = val - array[i]; array[i] = val; for (; i < array.size(); ++i) allSum[i] += tmp; } int sumRange(int i, int j) { if (i < 0 || i >= array.size() || j<0 || j >= array.size() || i>j) return 0; if (0 == i) return allSum[j]; else return allSum[j] - allSum[i - 1]; }private: vector<int> array; vector<int> allSum;};//树状数组实现class NumArray {private: vector<int> c; vector<int> m_nums;public: NumArray(vector<int> &nums) { c.resize(nums.size() + 1); m_nums = nums; for (int i = 0; i < nums.size(); i++){ add(i + 1, nums[i]); } } int lowbit(int pos){ return pos&(-pos); } void add(int pos, int value){ while (pos < c.size()){ c[pos] += value; pos += lowbit(pos); } } int sum(int pos){ int res = 0; while (pos > 0){ res += c[pos]; pos -= lowbit(pos); } return res; } void update(int i, int val) { int ori = m_nums[i]; int delta = val - ori; m_nums[i] = val; add(i + 1, delta); } int sumRange(int i, int j) { return sum(j + 1) - sum(i); }};
GitHub测试程序源码
- LeetCode(307) Range Sum Query - Mutable
- [leetcode-307]Range Sum Query - Mutable(java)
- leetcode 307 : Range Sum Query - Mutable
- [LeetCode 307] Range Sum Query - Mutable
- Leetcode (307) Range Sum Query - Mutable
- [LeetCode]Range Sum Query - Mutable
- Range Sum Query - Mutable -leetcode
- leetcode Range Sum Query - Mutable
- Leetcode: Range Sum Query - Mutable
- Leetcode: Range Sum Query - Mutable
- leetcode, Range Sum Query - Mutable
- LeetCode Range Sum Query - Mutable
- leetcode - Range Sum Query - Mutable
- Range Sum Query - Mutable -- Leetcode
- leetcode Range Sum Query-Mutable
- 【Leetcode】Range Sum Query - Mutable
- (LeetCode 307) Range Sum Query - Mutable(Segment Tree)
- (LeetCode 307) Range Sum Query - Mutable(树状数组讲解)
- 117 php __autoload 和 spl_autoload_register
- ROS学习笔记二:ubuntu软件源配置(二)
- java web遇到的各种问题汇总
- POJ 3253 哈夫曼树
- el表达式跟ognl表达式的区别(用法和页面取值)
- LeetCode(307) Range Sum Query - Mutable
- imx6 板卡移植官方yocto版本(1_安装环境)
- C/C++知识点
- NSString中initWithString和initWithFormat的区别
- 软件测试目的
- uclinux 与 linux的区别
- oc集合类的创建
- 函数重用和函数引用结合测试
- git 常用命令整理