ruby - 最大子数组 - 正确格式化输出

Question

我正在尝试经典的最大子数组问题。我正在输出正确的数组，但我希望它的格式不同。这是我的代码:

def max_subarray(input)
  return input if input.length == 1

  max = 0
  subarrays = []

  input.each do |element|
    current_subarray = []
    if max + element > element
      current_subarray << subarrays[-1]
      current_subarray << element
      subarrays << current_subarray
      max += element
    else
      subarrays << [element]
      max = element
    end
  end

  subarrays.each do |s|
    print s
    puts ''
  end
end

max_subarray([-2, -3, 4, -1, -2, 1, 5, -3])

这给了我以下输出:

[-2]
[-3]
[4]
[[4], -1]
[[[4], -1], -2]
[[[[4], -1], -2], 1]
[[[[[4], -1], -2], 1], 5]
[[[[[[4], -1], -2], 1], 5], -3]

虽然这些是我想要的数字，但我希望它们成为每一行的一个数组。不是嵌套数组。有什么想法吗？

谢谢，

附注一旦我把这部分记下来，找到具有最大总和的部分就是一小步。

Answer 1

我可能误解了你的问题，但你可能正在寻找 flatten。

...
subarrays.each do |s|
  print s.flatten
  puts ''
end
...

结果是:

(nick@monster)-(~/Desktop)
(504)⚡️ ruby derp.rb 
[-2]
[-3]
[4]
[4, -1]
[4, -1, -2]
[4, -1, -2, 1]
[4, -1, -2, 1, 5]
[4, -1, -2, 1, 5, -3]