求1~n中和为sum的所有组合(好未来笔试题)dfs

链接:https://www.nowcoder.com/questionTerminal/11cc498832db489786f8a03c3b67d02c
来源:牛客网

输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来

输入描述:

每个测试输入包含2个整数,n和m

输出描述:

按每个组合的字典序排列输出,每行输出一种组合
输入 : 6 8
输出:1 2 5
     1 3 4
     2 6
     3 5
     8

代码:深度优先搜索

#include<iostream>
#include<vector>
using namespace std;

void dfs(int n, int sum, vector<int>& res, int beg){//beg表示开始的位置
	if (sum == 0){
		for (int i = 0; i < res.size(); i++){
			if (i == 0){
				cout << res[i];
			}
			else{
				cout << " " << res[i];
			}
		}
		cout << endl;
	}
	for (int i = beg; i <= n&&i <= sum; i++){
		res.push_back(i);
		dfs(n, sum - i, res, i + 1);
		res.pop_back();//对于查找过得元素进行出
	}
}

int main()
{
	int sum, n;
	cin >> n >> sum;
	vector<int>res;
	dfs(n, sum, res, 1);
	//getchar();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/liugg2016/article/details/82118517