昨天看了一道子集和问题,回忆起了回溯法递归版的框架,然后开始试着组合数,也就是C(n,k) 一开始觉得这个是数学最基本的式子,但是实现起来好像不是那么好弄
看到子集和于是发现其实就是遍历,然后每次选或不选,然后分别递归进去,到头了就返回
设计递归函数参数,一个n遍历整个数组,k表示当前选了多少数字了,我习惯从后往前,因为到0 或者1 就递归出口,如果前向后会判断和 n k 是否相等,可能会加递归参数之类的。
但是调试还是遇到一个问题,之前写的n=0 就return, 结果发现 C(4,2) 总是输不出4 1 后来想了想加上debug发现了问题,我如果把n=0 return 写在前面就把 选到 41 最后算C(0 0)这种正确情况给错误的拦截了,因为当我遍历完这个数组后,最后是 0个选0个,所以n=0 的递归出口要写在后面, 仅仅处理一开始递归 参数从0个里选多少个的情况。另外还有 n<k 一定是不能算的,这个不符合定义,没有意义,可以写在k=0 前面
虽然写好了,但是看下leetcode这道题目,居然是需要返回把组合数作为vector返回的,搞了个递归不能用,还是要写个非递归版的回溯算法,不过这个先写在这里,可以和后面的对比
#include <vector>
#include <iostream>
using namespace std;
vector<vector<int>> vector0;
vector<int&