suseoj 1211: 子集和问题 (dfs)

1211: 子集和问题

时间限制: 1 Sec  内存限制: 128 MB
提交: 2  解决: 2
[提交][状态][讨论版][命题人:liyuansong]

题目描述

子集和问题的一个实例为<S，t>。其中，S={x₁，x₂，...，x_n}是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S₁，使得

试设计一个解子集和问题的回溯法。

对于给定的正整数的集合S和正整数c，计算S的一个子集S₁使得

输入

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

输出

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

样例输入

5 10
2 2 6 5 4

样例输出

2 2 6

C/C++ 代码实现（AC）：

 #include <iostream>
 #include <algorithm>
 #include <cstring>
 #include <cstdio>
 #include <cmath>
 #include <stack>
 #include <map>
 #include <queue>
 #include <climits>

 using namespace std;
 const int MAX = 1e6 + ;
 int A[MAX], ANS[MAX], n, m, N, flag = , my_book[MAX] = {};

 void dfs(int cnt, int step)
 {
     if (cnt == m)
     {
         N = step;
         flag = ;
         return ;
     }
     if (cnt > m) return ;
     for (int i = ; i < n; ++ i)
     {
         if (my_book[i]) continue;
         if (flag) return ;
         my_book[i] = ;
         ANS[step] = A[i];
         dfs(cnt + A[i], step + );
         my_book[i] = ;
     }
 }

 int main()
 {
     cin >>n >>m;
     for (int i = ; i < n; ++ i)
     {
         scanf("%d", &A[i]);
     }
     dfs(, );
     if (!flag) cout <<"Solution!" <<endl;
     else
     {
         for (int i = ; i < N - ; ++ i)
             cout <<ANS[i] <<" ";
         cout <<ANS[N - ] <<endl;
     }
     return ;
 }