子集和问题

子集和问题

Time Limit: 1000 ms Memory Limit: 65536 KiB

Submit Statistic

Problem Description

子集和问题的一个实例为〈S,t〉。其中，S={  x1 ， x2 ，…，xn }是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S1，使得：
。
试设计一个解子集和问题的回溯法。
对于给定的正整数的集合S={  x1 ， x2 ，…，xn }和正整数c，计算S 的一个子集S1，使得：
。

Input

输入数据的第1 行有2 个正整数n 和c（n≤10000，c≤10000000），n 表示S 的大小，c是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。

Output

将子集和问题的解输出。当问题无解时，输出“No Solution!”。

Sample Input

5 10
2 2 6 5 4

Sample Output

2 2 6

Hint

 

Source

#include <bits/stdc++.h>
using namespace std;

int s[10010], x[10010];
int n, c;

bool f(int n)
{
    int i = 0, sum = 0;
    while(i >= 0)
    {
        if(!x[i])
        {
            x[i] = true;
            sum += s[i];
            if(sum == c)
                return true;
            else if(sum > c)
            {
                x[i] = false;
                sum -= s[i];
            }
            i++;
        }
        if(i >= n)
        {
            while(x[i-1])
            {
                i--;
                x[i] = false;
                if(i < 1)
                    return false;
            }
            while(!x[i-1])
            {
                i--;
                if(i < 1)
                    return false;
            }
            sum -= s[i-1];
            x[i-1] = false;
        }
    }
    return false;
}

int main()
{
    cin>>n>>c;
    for(int i = 0; i < n; i++)
    {
        cin>>s[i];
    }
    if(f(n))
    {
        int flag = 1;
        for(int i = 0; i < n; i++)
        {
            if(x[i])
            {
                if(flag)
                    flag = 0;
                else
                    cout<<" ";
                cout<<s[i];
            }
        }
        cout<<endl;
    }
    else
    {
        cout<<"No Solution!"<<endl;
    }
    return 0;
}