ruby - 数组部分的总和

Question

我有一个数字数组xs，我需要计算其中三个连续部分的元素总和。这些部分是xs[0]，xs[1, N]并且xs[N + 1, N]对于一些整数N。例如，对于N = 2：

[1, 2, 3, 4, 5] #=> 1, 5, 9

我的实现如下。我无法提高内存/速度效率，但它仍然困扰着我，我想我现在无事可做，但也许有更优雅和简洁的方法来做到这一点？我不必产生一个数组，它可以是任何其他结构。

xs.each_with_index.reduce([0, 0, 0]) do |m, x|
  v, i = x
  j = (i > N ? 2
       : i > 0 ? 1
       : 0)
  m[j] += v.to_i
  m
end

Answer 1

[xs[0..0],xs[1..N],xs[N+1..-1]].map{|x|x.reduce{|a,b|a+b}}

a.rb：

N = (rand * 1E6).to_i
xs = 1.upto(1E6).to_a

xs.each_with_index.reduce([0, 0, 0]) do |m, x|
  v, i = x
  j = (i > N ? 2
       : i > 0 ? 1
       : 0)
  m[j] += v.to_i
  m
end

b.rb：

N = (rand * 1E6).to_i
xs = 1.upto(1E6).to_a

[xs[0..0],xs[1..N],xs[N+1..-1]].map{|x|x.reduce{|a,b|a+b}}

时间：

$ time ruby a.rb 
ruby a.rb  0.85s user 0.01s system 99% cpu 0.860 total

$ time ruby b.rb 
ruby b.rb  0.29s user 0.01s system 99% cpu 0.296 total

红宝石版本：

$ ruby -v
ruby 1.9.3p194 (2012-04-20 revision 35410) [x86_64-darwin12.0.0]

Answer 2

小变种：

[xs[0..0],xs[1..N],xs[N+1..-1]].map{|x|x.inject(:+)}