回溯法 实现组合数 从N个数中选择M个数

前言

    在平时的算法的题目中,时常会遇到组合数相关的问题,暴力枚举。在N个数中挑选M个数出来。利用for循环也可以处理,但是可拓展性不强,于是写这个模板供以后参考。

       两个函数和全局变量可以直接用。

代码:

#include<iostream>
#include<cstdio>

#define N 10    //被选择的数目
#define M 5    //要选出来的数目

using namespace std;
int vis[N+1];    //标志,
int ans=0;    //含有的组合数 的数量
int num[M+1];    //选出来的数放在num数组里面

void solve() {        //在solve函数里面处理
	for(int i=1; i<M+1; i++)
		cout<<num[i]<<" ";
	cout<<endl;
}

void dfs(int index) {    //挑选的第index+1个数
	if(index == M) {
		solve();
		ans++;	
		return ;
  • 3
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这个问题可以用回溯法来解决。回溯法是一种通过穷举所有可能的候选解来找出所有的解的法。具体来说,在回溯法,我们首先选择一个候选解,然后递归地搜索剩余的候选解,直到找到所有的解或者确定不存在解为止。 下面是一个使用回溯法来解决这个问题的 Python 代码实现: ```python def combinations(n, m): def backtrack(start, path): if len(path) == m: res.append(path[:]) return for i in range(start, n+1): path.append(i) backtrack(i+1, path) path.pop() res = [] backtrack(1, []) return res ``` 这个函的输入是两个整 `n` 和 `m`,表示在 1 到 `n` 之间选出 `m` 个数组合。函的输出是一个列表,包含所有可能的组合方式。函数中使用了一个内部函 `backtrack`,它负责递归地搜索候选解。在搜索过程,我们维护了一个 `path` 列表,它存储了当前已选的字。如果 `path` 的长度等于 `m`,那么就将 `path` 添加到结果列表 `res` 。否则,我们从 `start` 开始枚举所有可能的字,并将其添加到 `path` ,然后递归地搜索剩余的字。搜索完成后,需要将 `path` 最后添加的字弹出,以便进行下一次搜索。 下面是一个例子,演示了如何使用 `combinations` 函来输出所有可能的组合方式: ```python >>> combinations(4, 2) [[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]] >>> combinations(5, 3) [[1, 2, 3], [1, 2, 4], [1, 2, 5], [1, 3, 4], [1, 3, 5], [1, 4, 5], [2, 3, 4], [2, 3, 5], [2, 4, 5], [3, 4, 5]] ``` 在这个例子,`combinations(4, 2)` 输出了 1 到 4 选出 2 个数字的所有组合方式,`combinations(5, 3)` 输出了 1 到 5 选出 3 个数字的所有组合方式。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值