僅顯示總和小於或等於目標數的兩個排列 arrays 的組合

Question

我有兩個 arrays： teams = [1,2,3]和drivers = [4,5,6] 。 使用排列，我設法顯示了兩個 arrays 的所有組合，但設法定義了我想從每個數組中使用的值的數量。 因此，在“團隊”中我使用了 1 個值，在“驅動程序”中我使用了兩個。 我只想顯示總和小於或等於 10 的組合，並刪除所有重復項。

    teams = [1,2,3]
    drivers = [4,5,6]
    team = teams.permutation(1).to_a
    driver = drivers.permutation(2).to_a
    array = team.product(driver)
    target = 11

這是使用 1 個來自團隊的數字和 2 個來自司機的數字成功輸出兩個 arrays 的所有組合，如下所示：

[[1], [4, 5]], [[1], [4, 6]], [[1], [5, 4]], [[1], [5, 6]], [[1], [6, 4]], [[1], [6, 5]], [[2], [4, 5]],等等...

要僅顯示小於或等於 10 的值，我的預期結果將是： [[1], [4, 5]], [[1], [5, 4]],

然后沒有重復會讓我只剩下： [[1], [4, 5]]

我嘗試添加以下代碼行，但出現未定義的方法“<=”錯誤：

@array = array[0].product(*array[1..-1]).select { |a| a.reduce(:+) <= target }

我也試過這個但沒有運氣：

result = array.combination(1).select{|combi| combi.sum <= target}

@array = result

我猜這與排列有關？

Answer 1

teams = [1,2,3]
drivers = [2,5,4,5,6,4,5,7]
max_driver_sum = 10

我假設drivers可以包含重復元素（如我的示例），但我將在最后解釋如果沒有重復元素，計算將如何簡化。

---

作為第一步，讓我們在重復值和不重復值之間划分drivers 。

counts = drivers.tally
  #=> {2=>1, 5=>3, 4=>2, 6=>1, 7=>1}

dup_drivers, uniq_drivers = counts.partition { |_d,n| n > 1 }
                                  .map { |arr| arr.map(&:first) }​
  #=> [[5, 4], [2, 6, 7]]

所以，

dup_drivers
  #=> [5, 4]

uniq_drivers    
  #=> [2, 6, 7]

請參閱Enumerable#tally和Enumerable#partition 。

這里，

counts.partition { |_d,n| n > 1 } 
 #=> [[[5, 3], [4, 2]], [[2, 1], [6, 1], [7, 1]]]

---

首先計算兩個驅動程序相等的唯一組合：

dup_combos = teams.each_with_object([]) do |t,arr|
  max_driver = (max_driver_sum - t)/2
  dup_drivers.each do |d|
    arr << [[t],[d,d]] if d <= max_driver
  end
end
  #=> [[[1], [4, 4]], [[2], [4, 4]]]

---

接下來，計算兩個驅動程序不相等的唯一組合：

all_uniq = uniq_drivers + dup_drivers
  #=> [2, 6, 7, 5, 4]

all_uniq_combos = all_uniq.combination(2).to_a
  #=> [[2, 6], [2, 7], [2, 5], [2, 4], [6, 7], [6, 5],
  #    [6, 4], [7, 5], [7, 4], [5, 4]]

uniq_combos = teams.each_with_object([]) do |t,arr|
  adj_driver_sum = max_driver_sum - t
  all_uniq_combos.each do |combo|
    arr << [[t],combo] if combo.sum <= adj_driver_sum
  end
end
  #=> [[[1], [2, 6]], [[1], [2, 7]], [[1], [2, 5]], [[1], [2, 4]],
  #   [[1], [5, 4]], [[2], [2, 6]], [[2], [2, 5]], [[2], [2, 4]],
  #   [[3], [2, 5]], [[3], [2, 4]]]

參見數組#combination 。

---

最后一步是組合兩組組合：

a1 = dup_combos + uniq_combos
  #=> [[[1], [4, 4]], [[2], [4, 4]], [[1], [2, 6]], [[1], [2, 7]],
  #    [[1], [2, 5]], [[1], [2, 4]], [[1], [5, 4]], [[2], [2, 6]],
  #    [[2], [2, 5]], [[2], [2, 4]], [[3], [2, 5]], [[3], [2, 4]]]

排序后，這個結果如下。

a1.sort
  #=> [[[1], [2, 4]], [[1], [2, 5]], [[1], [2, 6]], [[1], [2, 7]],
  #    [[1], [4, 4]], [[1], [5, 4]],
  #    [[2], [2, 4]], [[2], [2, 5]], [[2], [2, 6]], [[2], [4, 4]],
  #    [[3], [2, 4]], [[3], [2, 5]]]

請注意， Array#uniq沒有在前面使用。 如果需要，當然可以替換掉上面的一些變量。

---

如果drivers不包含重復項，則所需數組由uniq_combos給出，其中all_uniq在all_uniq_combos的計算中被drivers替換。 例如，如果

teams = [1,2,3]
drivers = [2,5,4,6,7]
max_driver_sum = 10

然后

all_uniq_combos = drivers.combination(2).to_a
  #=> [[2, 5], [2, 4], [2, 6], [2, 7], [5, 4], [5, 6],
  #    [5, 7], [4, 6], [4, 7], [6, 7]]

combos = teams.each_with_object([]) do |t,arr|
  adj_driver_sum = max_driver_sum - t
  all_uniq_combos.each do |combo|
    arr << [[t],combo] if combo.sum <= adj_driver_sum
  end
end ​
  #=> [[[1], [2, 5]], [[1], [2, 4]], [[1], [2, 6]], [[1], [2, 7]],
  #    [[1], [5, 4]], [[2], [2, 5]], [[2], [2, 4]], [[2], [2, 6]],
  #    [[3], [2, 5]], [[3], [2, 4]]]

combos.sort
  #=> [[[1], [2, 4]], [[1], [2, 5]], [[1], [2, 6]], [[1], [2, 7]],
  #    [[1], [5, 4]],
  #    [[2], [2, 4]], [[2], [2, 5]], [[2], [2, 6]],
  #    [[3], [2, 4]], [[3], [2, 5]]]

Answer 2

這是一種方法

teams = [1, 2, 3]
drivers = [2, 6, 5, 4]
team = teams.permutation(1).to_a
driver = drivers.permutation(2).to_a
array = team.product(driver)

target = 10

res = array.select {|i| i.map(&:sum).sum <= target}.compact
==> [[[1], [2, 6]], [[1], [2, 5]], [[1], [2, 4]], [[1], [6, 2]],
     [[1], [5, 2]], [[1], [5, 4]], [[1], [4, 2]], [[1], [4, 5]],
     [[2], [2, 6]], [[2], [2, 5]], [[2], [2, 4]], [[2], [6, 2]],
     [[2], [5, 2]], [[2], [4, 2]], [[3], [2, 5]], [[3], [2, 4]],
     [[3], [5, 2]], [[3], [4, 2]]]

獲得獨特的物品（修改后也適用於團隊>車手的價值）

t1 = res.map {|i| i[0]}
d2 = res.map {|i| i[1].flatten.sort}

t1.zip(d2).uniq
==> [[[1], [2, 6]], [[1], [2, 5]], [[1], [2, 4]], [[1], [4, 5]],
     [[2], [2, 6]], [[2], [2, 5]], [[2], [2, 4]], [[3], [2, 5]],
     [[3], [2, 4]]]