循环没有将所有解决方案记录到数组中

Question

此循环搜索stock_picker 变量，并通过检查哪两天的净利润最高来确定哪两天是最好的“买入”和“卖出”天。循环正确地看到，在第 0 天买入并在第 1 天卖出，我的最大利润将是 8。

但是，我希望程序将所有可能的最佳买卖日期记录到一个数组中。如果我在第 0 天买入并在第 3 天卖出，我的利润仍然是 8，但程序没有记录这一点。相反，它返回一个 [0,1,0,1] 数组，告诉我它看到了两个解决方案，但由于某种原因没有记录第二个解决方案。如何返回 [0,1,0,3] 的数组？

def stock_picker(prices)
  buy_and_sell_days = []

  best_profit = 0

  prices.each do |low|
    prices.each do |high|
      if prices.index(high) > prices.index(low)
        profit = high - low
        if profit > best_profit
          best_profit = profit
        end
        if high - low == best_profit
          buy_and_sell_days.push(prices.index(low), prices.index(high))
        end
      end
    end
  end

  p buy_and_sell_days
  p best_profit

end

stock_picker([1, 9, 2, 9])

（为格式/易读性而编辑）

Answer 1

index 方法：

返回 ary 中第一个对象的索引，使得该对象 == 到 obj。

所以：prices.index(1) 是 0，prices.index(9) 是 1

如果你使用 index 方法，它没有足够的信息知道选择第二个而不是第一个，所以你得到[0, 1, 0, 1]。

您可以将索引与它一起传递，这样您就可以完全避免这种情况。您还可以存储所有可能的利润和日期，然后在完成后获取最大值。更轻松！这是两个：

def stock_picker(prices)
  buy_and_sell_days = []
  buy_and_sell_days2 = {}

  best_profit = 0

  prices.each_with_index do |low, lidx|
    prices.each_with_index do |high, hidx|
      if hidx > lidx
        buy_and_sell_days2[high - low] ||= []
        buy_and_sell_days2[high - low] << [lidx, hidx]
        profit = high - low
        if profit > best_profit
          best_profit = profit
        end
        if high - low == best_profit
          buy_and_sell_days.push(lidx, hidx)
        end
      end
    end
  end

  p buy_and_sell_days
  p buy_and_sell_days2
  p buy_and_sell_days2.max_by { |profit, pairs| profit }
  p best_profit
end

stock_picker([1,9,2,9])
#> [0, 1, 0, 3]
#> {8=>[[0, 1], [0, 3]], 1=>[[0, 2]], -7=>[[1, 2]], 0=>[[1, 3]], 7=>[[2, 3]]}
#> [8, [[0, 1], [0, 3]]]
#> 8