输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来.
#include "stdafx.h"
#include <iostream>
using namespace std;
int length;
void PrintSolutions(int *flag)
{
for (int i=0; i<length; i++)
{
if (flag[i] == 1)
{
cout << i+1 << " ";
}
}
cout << endl;
}
void BagProblem(int m, int n, int *flag)
{
if(n<1 || m<1)
return;
if(m < n)
n = m;
if (n == m)
{
flag[n-1] = 1;
PrintSolutions(flag);
flag[n-1] = 0;
}
flag[n-1] = 1;
BagProblem(m-n, n-1, flag);
flag[n-1] = 0;
BagProblem(m, n-1, flag);
}
int main(int argc, char* argv[])
{
int m, n;
cout << "Please input the m and n:" << endl;
cin >> m >> n;
cout << "The solution is:" << endl;
length = n;
int *flag = (int *)malloc(sizeof(int)*n);
memset(flag, 0, sizeof(flag));
BagProblem(m,n,flag);
//delete flag;//这个地方犯了一个原则性的错误 new和delete成对使用, malloc应该和free成对使用,要不然就会造成内存泄露
free(flag);
return 0;
}
上面的代码看懂不是很难吧,递归的程序就是可读性好,我觉得这个代码的亮点应该就是flag数组的使用,充分利用了递归的性质,只是很简单的一个数组就完成了所有组合的输出。在每次把flag[i]设置为1之后就进入递归,代表了将i放入背包,当退出递归函数的时候,肯定要将flag[i]赋为0,因为这时候i已经不在背包中了。