Ruby [更新方法] 怎么样用比较优美的方法得到某一数组的累加数组，例如 [1,3,5,7] => [1,4,9,16]
怎么样用一个比较优美的方法计算得到一个累加数组，如下
#测试用例：
[] => []
[0, -3, 5, 7] => [0, -3, 2, 9]
[1, 3, 5, 7] => [1, 4, 9, 16]
谢谢帮忙，先总结一下大家目前的劳动成果
a = [1,3,5,7]
# Method 1 from @fredwu @abitno => O(n)
a.inject([]) { |sum,e| sum << (sum[-1] || 0) + e }
# Method 2 from @luikore => O(n)
sum = 0
a.map {|e| sum += e }
# Method 3 from @nil => O(n^2)
a.length.times.map {|i| a[0..i].inject(:+)}
# Update Method 3 => O(n)
temp = 0
a.length.times.map {|i| temp = [temp, a[i]].inject(:+)}
//////////////////// 更新 /////////////////////// 第三种方法的好处是可以扩展为其他运算符号，这样累加或者累乘都可以解决了。
不过缺点是计算效率好像不怎么高，每次都是从头开始算，前面两个算法都是 O(n), 第三个方法是 O(n^2)，对于一般的数组基本上没啥区别，如果是大数组，就要考虑下了。
给这个方法取个名字吧，想到了 Accumulation 这个单词，查了下 webster 词典，该词的英文意思是 Accumulation - increase or growth by addition especially when continuous or repeated 应该还算贴切，而且 acum 和 sum 还挺押韵的
def acum(a, init, &oper)
a.length.times.map {|i| a[0..i].inject(init, &oper)}
end
#测试用例
a = [1,3,5,7]
#累加（注意init应设为0）
p acum(a, 0, &:+) => [1, 4, 9, 16]
#累乘（注意init应设为1）
p acum(a, 1, &:*) => [1, 3, 15, 105]
也可以用在字符数组的累积合并中
a = %w[r u b y]
#字符串的累积合并
p acum(a, "", &:+) => ["r", "ru", "rub", "ruby"]
//////////////////// 更新 /////////////////////// 简单测试了下，对大数组，两种方法区别还是挺大的
class Array
def acum_add
sum = 0
self.map {|e| sum += e }
end
def acum(init=0, &oper)
self.length.times.map {|i| self[0..i].inject(init, &oper)}
end
end
(1..100).to_a.acum_add # => [Finished in 0.2-0.3s]
(1..100).to_a.acum(&:+) # => [Finished in 0.2-0.3s]
(1..10000).to_a.acum_add # => [Finished in 0.2-0.4s]
(1..10000).to_a.acum(&:+) # => [Finished in 8.8s] !!!!!
//////////////////// 更新 /////////////////////// 把 Method 3 改进了下，现在也是 O(n) 了，各取所长，不知道有没有到 O(logn) 的
class Array
def acum(init=0, &oper)
temp = init
self.length.times.map {|i| temp = [temp, self[i]].inject(init, &oper)}
end
end
(1..10000).to_a.acum(&:+) # => [Finished in 0.2-0.4s]
a = [1,3,5,7]
b=[]; a.inject(0){ |r, e| b << r+e; b[-1] }; b #=> [1, 4, 9, 16]
[1,3,5,7].inject([]) {|result, el| result << el + (result[-1] || 0) }
[1,2,4].inject([]) {|result, el| result << el + (result[-1] || 0) }
我觉得写成 1 行未必就是优美的，楼上几个都是用了动态规划，从底向上计算，日后阅读可能有困难。我写个罗嗦版本的 memoization。
def memoize(f)
cache = {}
lambda {|*args| cache[args] ||= f.call(*args) }
end
sum_to_index = lambda do |array, index|
if index == 0
array[index]
else
array[index] + sum_to_index.call(array, index - 1)
end
end
sum_to_index = memoize(sum_to_index)
array = [1, 3, 5, 7]
array.map.with_index {|entry, index| sum_to_index.call(array, index) }
#11 楼 @fresh_fish 也许吧，哈哈
很多时候看 legacy code，都是不知道代码在解决什么问题，需要猜才可以。 看到这些 one liner 我都是要想一下才知道是做什么的。 如果一定要写 one liner，最好也封装到一个方法里，这样看代码的人至少知道这是干嘛的。
a = [1,3,5,7] b=[] a.reduce([]){|result,element| b << (result.last || 0) + element} 不知道 reduce 和 inject 他们的具体实现那个更好
#17 楼 @qinjker 看了一篇文章有关 map 与 collect，reduce 与 inject 区别和历史由来 http://www.rubycc.com/bbs/topic_detail/125
一劳永逸：
[1,3,5,7].sum
class Array
def sum
self.reduce([]) { |sum,e| sum << (sum[-1] || 0) + e }
end
end