Как подойти к следующей проблеме жадных / динамических c? - PullRequest
0 голосов
/ 09 февраля 2020

Я должен проверить, можно ли получить сумму от заданных чисел> строго больше, чем заданная сумма

, например:

, если данная сумма равна 9, а заданные числа равны [3,5,7], тогда правильный ответ равен 1 * 3 + 1 * 7 = 10> 9, поэтому выходной результат должен быть [1,0,1]

example2:

если заданная сумма равна 13 и числа [1,3,5], тогда правильный ответ - 0 * 1 + 3 * 3 + 1 * 5 = 14> 13 но 9 * 1 + 0 * 3 + 5 = 14> неверно, поэтому правильный вывод [0,3,1], как подходить к этой проблеме

1 Ответ

0 голосов
/ 10 февраля 2020

Решение можно получить с помощью версии неограниченного алгоритма ранца.

Здесь мы модифицируем код из Python Несвязанный рюкзак

Изменить с помощью две цели:

  1. Минимизировано количество значений, используемых в массиве

  2. Создать максимальную сумму, меньшую, чем предел (т.е. данная сумма + 1)

def knapsack_unbounded_dp(arr, C):
    C += 1  # actual limit

    # Form index value pairs (to keep track of the original indexes after sorting)
    items = [(i, v) for i, v in enumerate(arr)]

    # Sort values in descending order
    items = sorted(items, key=lambda item: item[1], reverse=True)

    # Sack keeps track of:
    #  max value so i.e. sack[0] and
    # count of how many each item are in the sack i.e. sack[1]
    sack = [(0, [0 for i in items]) for i in range(0, C+1)]   # value, [item counts]

    for i,item in enumerate(items):
        # For current item we check if a previous entry could have done better
        # by adding this item to the sack
        index, value = item
        for c in range(value, C+1):   # check all previous values
            sackwithout = sack[c-value]  # previous max sack to try adding this item to
            trial = sackwithout[0] + value  # adding value to max without using it
            used = sackwithout[1][i]        # count of i-them item
            if sack[c][0] < trial:
                # old max sack with this added item is better
                sack[c] = (trial, sackwithout[1][:])
                sack[c][1][i] +=1   # use one more

    value, bagged = sack[C]

    # index and count of each array value
    new_bagged = [(i, v) for (i, _), v in zip(items, bagged)]

    # Re-sort based upon original order of array indexes
    new_bagged.sort(key=lambda t: t[0])

    # counts based upon original array order
    cnts = [v for i, v in new_bagged]

return sum(cnts), value, cnts

Тестовый код

for t in [([1, 3, 5], 8), ([3, 5, 7], 9), ([1, 3, 5], 13)]:
  result = knapsack_unbounded_dp(t[0], t[1])
  print(f'Test: {t[0]}, Given Sum {t[1]}')
  print(f'Result counts {result[2]}, with max sum {result[1]}, Total Count {result[0]}\n')

Выход

Test: [1, 3, 5], Given Sum 8
Result counts [0, 3, 0], with max sum 9, Total Count 3

Test: [3, 5, 7], Given Sum 9
Result counts [0, 2, 0], with max sum 10, Total Count 2

Test: [1, 3, 5], Given Sum 13
Result counts [0, 3, 1], with max sum 14, Total Count 4
...