作者 王东
单位 贵州师范学院
给定n个不同的正整数集合w=(w1,w2,…,wn)和一个正数W,要求找出w的子集s,使该子集中所有元素的和为W。
输入格式:
第一行输入n和W,第二行依次输入n个数。
输出格式:
每行输出一个符合要求的子集。
输入样例1:
4 31
11 13 24 7
输出样例1:
11 13 7
24 7
个人思路:
回溯。
穷举每一种情况,符合条件时输出。
注意保护现场,及时恢复。
代码:
#include <bits/stdc++.h>
using namespace std;
int n,w;
const int N = 110;
int f[N];
vector<int> p;
void dfs(int u,int j) {
if (u == w) { //当子集和与w相等时就可以输出了
int len = p.size();
for (int i = 0; i < len; i++) {
cout << p[i]<<" ";
}
cout << endl;
return;
}
for (int i = 0; i < n; i++) {
if (u + f[i] <= w && i>j) {
p.push_back(f[i]);
u += f[i];
dfs(u,i);
p.pop_back(); //恢复现场
u -= f[i];
}
}
}
int main()
{
cin >> n >> w;
for (int i = 0; i < n; i++) {
cin >> f[i];
}
dfs(0,-1);
}