微软算法面试(16):输入两个整数n和m, 从数列1,2,...,n中任意选择几个数,使其和等于m, 要求编写程序输出所有的组合

编程求解:

输入两个整数 n
和 m,从数列1,2,3…n中随意取几个数,

使其和等于 m ,要求将其中所有的可能组合列出来.

分析:

可以使用递归思想, 从第n个数开始找其组合,

1)包括n的所有组合

2)不包括n的所有组合

把所有组合整合起来就OK了。

下面来看具体的实现:

#include<iostream>
#include<vector>
#include<set>
#include<math.h>
using namespace std;

void Getmsets(int n, int m, vector<vector<int> >&allsets)
{
	if(n < 1 || m < 1) return;
	if(m > n*(n+1)/2) return;//n too small
	if(n > m)
		n = m;
	vector<int> _vecset;
	if(n == m)
	{
		_vecset.push_back(n);
		allsets.push_back(_vecset);
		Getmsets(n-1, m, allsets);
	}
	else
	{
		vector<vector<int> > _vecvecsets;
		//包括n的情况
		Getmsets(n, m -n, _vecvecsets);
		vector<vector<int> >::iterator itvecvec;
		for(itvecvec = _vecvecsets.begin(); itvecvec != _vecvecsets.end(); ++ itvecvec)
		{
			itvecvec->push_back(n);
			allsets.push_back(*itvecvec);
		}
		_vecvecsets.clear();
		//不包括n的情况
		Getmsets(n-1, m, _vecvecsets);
		for(itvecvec = _vecvecsets.begin(); itvecvec != _vecvecsets.end(); ++ itvecvec)
		{
			allsets.push_back(*itvecvec);
		}
	}
} 

int main()
{
	
	vector<vector<int> > vecvecres;
	Getmsets(10, 10, vecvecres);
	cout << "n = 10, m = 10, 所有的组合为:" << endl;
	vector<vector<int> >::iterator itvecvec;
	for(itvecvec = vecvecres.begin(); itvecvec != vecvecres.end(); ++ itvecvec)
	{
		for(vector<int>::iterator itvec = itvecvec->begin(); itvec != itvecvec->end(); ++itvec)
		{
			cout << *itvec << ",";
		}
		cout << endl;
	}
	return 0;
}

输出结果为:

n = 10, m = 10, 所有的组合为:

10,

1,9,

2,8,

3,7,

1,2,7,

4,6,

1,3,6,

5,5,

1,4,5,

2,3,5,

2,4,4,

3,3,4,

1,2,3,4,

posted @ 2021-01-30 22:53  天涯学馆  阅读(463)  评论(1编辑  收藏  举报