打印在C ++中与目标相加的子集

时间:2014-02-08 21:11:02

标签: c++ algorithm runtime

我正在尝试实现一种算法,该算法打印一个数组的子集,该数组在C ++中与目标相加。我从互联网上获得了算法并用C ++实现了它。有人可以找出为什么我无法打印完整的输出?同样在复杂性方面,我认为这个算法是O(2 ^ N)。还有其他更好的复杂算法吗?

这是我的代码:

void printSubsets_sumto_target(vector<int> a,int k,vector<int> partial)
{
    int s =0;
    int i,j;
    int psize = partial.size();
    int osize = a.size();
    vector<int> remaining;
    vector<int> partial_rec;

    for(i=0;i<psize;i++)
        s+=partial[i];
    //cout<<"Sum : "<<s<<endl;
    if(s==k)
    {
        //cout<<"Partial size : "<<psize<<endl;
        for(i=0;i<psize;i++)
            cout<<partial[i]<<", ";
        cout<<endl;
    }
    if(s>=k)
        return;
    for(i=0;i<osize;i++)
    {
        remaining.empty();
        int n = a[i];
        for(j=i+1;j<osize;j++)
            remaining.push_back(a[j]);
        partial_rec.empty();
        partial_rec.push_back(n);
        printSubsets_sumto_target(remaining,k,partial_rec);
    }
}

这个算法和一个向量a = {1,2,3,8,9,7}我用

调用
vector<int> partial;
printSubsets_sumto_target(a,10,partial);

我得到的输出是

3, 7, 
2, 8, 
3, 7, 
3, 7, 
2, 8, 

预期输出

1,9
2,8
3,7

我哪里错了?

2 个答案:

答案 0 :(得分:1)

1。这是一个NP问题(但排序值可以减少很多组合)

2。如上所述,子集可以包含3个或更多元素,因此输出可以是7 2 1

3。我曾经写过练习代码,希望它会有所帮助

#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
int a[100]={1,2,3,8,9,7};
bool x[100];
int N=6;//number of elements
int t=10;//target sum
int sum;//current target sum
int cmp(const void *a,const void *b)
{
    return *(int *)b-*(int *)a;
}
void backtrace(int n)
{
    if(sum>t)
        return ;
    if(sum==t)
    {
        for(int j=0;j<n;++j)
        {
            if(x[j])
                cout<<a[j]<<" ";
        }
        cout<<endl;
        return;
    }
    if(n==N)
        return ;
    for(int i=n;i<N;++i)
    {
        if(x[i]==false)
        {
            x[i]=true;
            sum+=a[i];
            backtrace(i+1);
            x[i]=false;
            sum-=a[i];
            while(i<N-1 && a[i]==a[i+1])
                i++;
        }
    }
}
int main()
{
    sum=0;
    memset(x,0,sizeof(x));
    qsort(a,N,sizeof(a[0]),cmp);
    backtrace(0);
    return 0;
}

输出:

9 1

8 2

7 3

7 2 1

答案 1 :(得分:0)

以下是一种时间复杂度为O(M*N)的算法,而M为目标,N为集合的总大小。使用带背包问题的类比如下: -

  
      
  1. 背包容量=目标
  2.   
  3. 项目是集合中的元素,其中包含权重&amp;价值与自身相同
  4.   
  5. 使用动态编程计算最大利润
  6.   
  7. maxprofit =目标然后是/是总结为目标的子集。
  8.   
  9. 回归解决方案。
  10.   

Java解决方案: -

public class SubSetSum {
    static int[][] costs;

    public static void calSets(int target,int[] arr) {

        costs = new int[arr.length][target+1];
        for(int j=0;j<=target;j++) {
            if(arr[0]<=j) {

                costs[0][j] = arr[0]; 
            }
        }
        for(int i=1;i<arr.length;i++) {

            for(int j=0;j<=target;j++) {
                costs[i][j] = costs[i-1][j];
                if(arr[i]<=j) {
                    costs[i][j] = Math.max(costs[i][j],costs[i-1][j-arr[i]]+arr[i]);
                }
            }

        }

        System.out.println(costs[arr.length-1][target]);
       if(costs[arr.length-1][target]==target) {
           System.out.println("Sets :");
           printSets(arr,arr.length-1,target,"");
       } 

       else System.out.println("No such Set found");

    } 

    public static void printSets(int[] arr,int n,int w,String result) {


        if(w==0) {
            System.out.println(result);
            return;
        }

        if(n==0) {
           System.out.println(result+","+arr[0]);
            return; 
        }

        if(costs[n-1][w]==costs[n][w]) {
            printSets(arr,n-1,w,new String(result));
        }
        if(arr[n]<=w&&(costs[n-1][w-arr[n]]+arr[n])==costs[n][w]) {
            printSets(arr,n-1,w-arr[n],result+","+arr[n]);
        }
    }

    public static void main(String[] args) {
        int[] arr = {1,2,3,8,9,7};
        calSets(10,arr);
    }
}