Динамическое программирование: идеальная сумма с отрицательными числами - PullRequest
0 голосов
/ 19 сентября 2018

Учитывая массив целых чисел и сумму, задача состоит в том, чтобы напечатать все подмножества данного массива с суммой, равной данной сумме.

Example: 
Input : arr[] = {1, 2, 3, 4, 5}
        sum = 10
Output : [4 3 2 1]  
         [5 3 2] 
         [5 4 1]

Input : arr[] = {-1, 2, 3, 4, 5}
        sum = 10
Output : [5 3 2] 
         [5 4 2 -1]

Я сделал это, используя динамическое программирование в псевдополиномиальное время.Это расширение проблемы суммы подмножеств, которая только решает, существует ли такое подмножество или нет.Мое решение ниже работает как для положительных, так и для отрицательных чисел для задачи с суммой подмножеств.Тем не менее, он не может правильно печатать подмножества, если массив содержит отрицательные числа. Программа -

import java.util.ArrayList;

// sum problem
class GFG {

    static boolean subset[][];

    // Returns true if there is a subset of
    // set[] with sun equal to given sum
    static boolean isSubsetSum(int set[],
                               int n, int sum) {
        // The value of subset[i][j] will be
        // true if there is a subset of
        // set[0..j-1] with sum equal to i
        subset = new boolean[n + 1][sum + 1];

        // Fill the subset table in botton
        // up manner
        for (int i = 0; i <= n; i++) {
            for (int j = 0; j <= sum; j++) {
                if (j == 0) {
                    subset[i][j] = true;
                } else if (i <= 0 && sum >= 1)
                    subset[i][j] = false;
                else if (set[i - 1] > j)
                    subset[i][j] = subset[i - 1][j];
                else {
                    if (set[i - 1] >= 0)
                        subset[i][j] = subset[i - 1][j] || subset[i - 1][j - set[i - 1]];
                    else
                        subset[i][j] = subset[i - 1][j] || subset[i - 1][j + set[i - 1]];
                }
            }
        }

        // uncomment this code to print table
//        for (int i = 0; i <= sum; i++)
//        {
//        for (int j = 0; j <= n; j++)
//            System.out.println (subset[i][j]);
//        }

        return subset[n][sum];
    }

    /* Driver program to test above function */
    public static void main(String args[]) {
        int set[] = {1, 2, 3, 4, 5};
        int sum = 10;
        int n = set.length;
        if (isSubsetSum(set, n, sum) == true)
            System.out.println("Found a subset"
                    + " with given sum");
        else
            System.out.println("No subset with"
                    + " given sum");
        System.out.println("Done");
        ArrayList<Integer> list = new ArrayList<>();
        printSubsets(set, n, sum, list);
        System.out.println("Finished");
    }

    static void display(ArrayList<Integer> v) {
        System.out.println(v);
    }

    private static void printSubsets(int[] set, int i, int sum, ArrayList<Integer> list) {
        if (i == 0 && sum != 0 && subset[0][sum]) {
            list.add(set[i]);
            display(list);
            list.clear();
            return;
        }

        // If sum becomes 0
        if (i == 0 && sum == 0) {
            display(list);
            list.clear();
            return;
        }

        // If given sum can be achieved after ignoring
        // current element.
        if (subset[i - 1][sum]) {
            // Create a new vector to store path
            ArrayList<Integer> b = new ArrayList<>();
            b.addAll(list);
            printSubsets(set, i - 1, sum, b);
        }

        // If given sum can be achieved after considering
        // current element.

        if (sum >= set[i - 1] && subset[i - 1][sum - set[i - 1]]) {
            list.add(set[i - 1]);
            printSubsets(set, i - 1, sum - set[i - 1], list);
        }

    }   
} 

Как этот код можно изменить, чтобы он работал и для отрицательных чисел?

Ответы [ 3 ]

0 голосов
/ 26 января 2019

Вы можете вычесть минимальное отрицательное число массива из всего набора, делая числа в массиве положительными.Затем примените динамическое программирование.

0 голосов
/ 29 апреля 2019

В ваших решениях предполагается, что все значения положительны, поэтому массив динамического программирования subset заполнен положительными значениями j, но теперь вам необходимо учитывать отрицательные суммы.

Вам нужно изменить пределы цикла j для заполнения массива динамического программирования на

for (int j = negative_sum; j <= positive_sum; j++)

, где negative_sum - сумма всех отрицательных значений, а positive_sum - сумма.из всех положительных.

Для получения более подробной информации читайте страницу википедии по проблеме суммы подмножеств здесь , где объясняется этот шаг.

0 голосов
/ 19 сентября 2018

Так как вам нужно напечатать (или сгенерировать) все возможные подмножества данного набора (содержащие как положительные, так и отрицательные целые числа), которые имеют суммирование, равное сумма , что вы можете сделать:

попытаться представить каждую позицию набора в виде двоичного представления 0 и 1, где 1 указывает, что элемент в этой позиции занят, а 0 указывает, что элемент в этой позиции не принимаетсяaccount.

Найдите суммирование всех позиций, где есть 1. Если сумма этих значений точно равна заданной сумме, выведите это подмножество.

Итак, общая сложность времени равнаO(2 ^ n), где n - длина данного набора.

Вы можете посмотреть на следующую реализацию.

import java.util.Arrays;

public class PerfectSum {

public static void printSubsets(int[] set, int n, int sum) {
     int totalSubSets = (1 << n);
     for (int i = 1; i < totalSubSets; ++i) { // loop over all possible subsets
         int curSum = 0;
         for (int j = n - 1; j >= 0; --j) {
             if (((i >> j) & 1) > 0) { // if bit at jth position is 1 take that value
                curSum +=set[j];
             }
         }
         if (curSum == sum) { // valid subset found, then print it
             for (int j = n - 1; j >= 0; --j) { // looping in reverse order to print set in decreasing order
                 if (((i >> j) & 1) > 0) { // if bit at jth position is 1 take that value
                     System.out.print(set[j] + " ");
                 }
             }
             System.out.println("");
         }
     }
}

public static void main(String[] args) {
    int set[] = {-1, 2, 3, 4, 5};
    Arrays.sort(set); // To print in non increasing order
    int sum = 10;
    int n = set.length;
    printSubsets(set, n, sum);
  }
}
...