打印一组的子集,其总和等于k

时间:2015-02-21 06:16:48

标签: algorithm dynamic-programming subset-sum

动态编程提供了一种解决子集和问题的非常优雅的方法。子集和问题:查找是否存在sum = k的子集。

但是我无法看到如何用sum = k打印所有子集。有关如何修改以下基于动态编程的函数的任何指针,如果存在所需的子集,则仅检查并返回true。请参阅HERE了解更多详情。

// Returns true if there is a subset of set[] with sun equal to given sum
bool 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
    bool subset[sum+1][n+1];

    // If sum is 0, then answer is true
    for (int i = 0; i <= n; i++)
      subset[0][i] = true;

    // If sum is not 0 and set is empty, then answer is false
    for (int i = 1; i <= sum; i++)
      subset[i][0] = false;

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

    /* // uncomment this code to print table
     for (int i = 0; i <= sum; i++)
     {
       for (int j = 0; j <= n; j++)
          printf ("%4d", subset[i][j]);
       printf("\n");
     } */

     return subset[sum][n];
}

1 个答案:

答案 0 :(得分:0)

我们假设子集数组已经存在。

我们可以写一个递归函数来打印所有子集:generate(sum,prefixLength)。它打印前缀的所有子集,前缀为prefixLength元素,总计为总和。在里面,我们需要检查两件事:如果子集[sum - set [prefixLength]] [prefixLength - 1]为真,我们应该为(sum - set [prefixLength],prefixLength - 1)生成所有子集,追加当前元素他们每个人都返回结果。如果subset [sum] [prefixLength - 1]为真,我们应该返回为(sum,prefixLength - 1)生成所有集合的结果。有时这两个选项都是可能的(在这种情况下,我们需要返回第一个和第二个选项的并集)。答案是生成(sum,n)。