代码随想录算法训练营第24天|77. 组合

77.组合

思路:如果暴力解,需要几个数则需要相应的for循环个数。

回溯法:把数的组合抽象成一颗树,利用递归的思想进行回溯,递归必有回溯。每次遍历到叶子节点,则存入结果。

代码：

vector<vector<int>> result;//存放结果
    vector<int> path;//存放当前进行到的路径
    void backtracking(int n,int k,int startIndex){//startIndex控制起始位置到末尾的集合大小,逐步缩小集合
        if(path.size() == k){//当前遍历到了叶节点,把路径加入结果并且返回上一节点
            result.push_back(path);
            return;
        }
        for(int i = startIndex;i <= n;i++){//遍历当层横向的节点
            path.push_back(i);
            backtracking(n,k,i+1);
            path.pop_back();//回溯
        }
    }
    vector<vector<int>> combine(int n, int k) {
        result.clear();
        path.clear();
        backtracking(n,k,1);
        return result;
    }

剪枝优化:假如n=4,k=3。所以数组为1，2，3，4中选取3个数的所有组合。当第一个选取3的时候，当时后面仅有4可选,不满足3个数的组合条件，这种情况如上代码还会进行一步递归，所以这种情况是可以优化的。当起点选为2的时候,这是最后满足条件的起点,当在这一步的时候往后就不需要再递归遍历了。所以k-path.size()是当前还需要找几个元素,n-(k-path.size())如果为正说明后面的元素个数满足还需要找的元素个数。+1则是因为当前不包括startIndex。例如path中没有元素,size为0,则利用这个公式算出为2.指最多到2就可以停下遍历了，后面再遍历也不够元素个数。

for循环条件更改代码：

for(int i = startIndex;i <= n - (k - path.size()) + 1;i++){
            path.push_back(i);
            backtracking(n,k,i+1);
            path.pop_back();
        }