前言:
21考研,正在啃《算法笔记》,不论能否进复试记录一下准备路上写下的垃圾代码。
解答:
#include<iostream>
using namespace std;
int a[20];
int n = 1;
int count(int i, int sum)
{
if (sum == 0) { return 1; } //找到一组和为sum的组合数;
if (i == n || sum < 0) return 0;//i==n说明没有其他的数来组合,sum<0说明组合不出;
return count(i + 1, sum - a[i]) + count(i + 1, sum);//从数组的第i为开始,包含a[i],和不包含;
}
int main()
{
while (cin >> n) {
for (int i = 0; i < n; i++)
cin >> a[i];
cout << count(0, 40) << endl;
}
return 0;
}
这个解答是牛客网链接:神奇的口袋
评论区用户phonkivon的解答,递推式中
count(i,sum)表示从数组的第i个数开始往后统计的,各种组合数中体积和为sum的种类数,
sum为组合数中体积的和,则:
cout(i,sum)=cout(i+1,sum-a[i])+cout(i+1,sum),
对于第i个数字,其加入后的组合数由两部分组成,
cout(i+1,sum-a[i])表示包含了a[i],即为从第i+1个数开始往后统计组合数的和为sum-a[i]的种类数,
cout(i+1,sum)表示不包含a[i], 即为从第i+1个数开始往后统计组合数的和为sum的种类数
之前写过全排列,看到这道题思维就有点被限制了,一直在想从全排列的递归算法中改出这个题目的答案,但是想了很多方法都觉得很麻烦。
看了这个解法真是豁然开朗,递归真是太奇妙了,想要用好递归还要走很多的路。