回溯解决组合问题——leetcode77

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。
例：输入：n=4,k=2
输出：[[1,2],[1,3],[1,4],[2,3],[2,4],[3,4]]

思路：回溯法的经典问题

直接解法：使用循环，这里k=2,可以使用两个for循环，代码如下：

int n=3;
for(int i=1;i<=n;i++){
    for(int j=i+1;j<=n;j++){
        cout<<i<<" "<<j<<endl;
   }
}

k等于几就有几层for循环，所以，当k比较大时，for循环暴力搜索是写不出来的。因此，需要用到回溯法暴力搜。我们用递归来解决多层嵌套循环的问题。将问题抽象成树型结构后：

每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围。
图中发现N相当于树的宽度，k相当于树的深度。
而每次搜索到了叶子节点，就找到了一个结果。因此，只要把达到叶子节点的结果收集起来，就可以求得n个数中k个数的组合集合。
按照上节讲到的三部曲，确定递归函数的返回值和参数、终止条件、单层搜索逻辑。

• 返回值和参数
  需要定义两个全局变量，一个用来存放符合条件的单一结果，一个用来存放符合条件的集合。代码如下：

vector<vector<int>> result;  //存放符合条件结果的集合
vector<int> path;  //用来存放符合条件结果

且函数中一定需要两个参数，从集合n中取k的数，那么n和k是两个int型的参数。然后还需要一个int型变量startIndex，用来记录本层递归中，集合从哪里开始遍历。这是因为：每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围，需要startIndex。从下图中红线部分可以看出，在集合[1,2,3,4]取1之后，下一层递归，就要在[2,3,4]中取数了，那么下一层递归如何知道从[2,3,4]中取数呢，靠的就是startIndex。

那么整体代码如下：

vector<vector<int>> result; // 存放符合条件结果的集合
vector<int> path; // 用来存放符合条件单一结果
void backtracking(int n, int k, int startIndex)`

• 终止条件
  即什么时候达到叶子节点，如果path数组的大小达到k，说明找到了一个子集大小为k的组合，在图中path存的就是根节点到叶子节点的路径。如图红色部分：

此时用result数组，将path保存起来，并终止本层递归。所以终止代码如下：

if(path.size()==k){
   result.push_bcak(path);
   return;
}

• 单层搜索逻辑
  for循环每次从startIndex开始遍历，然后用path保存取到的节点i。代码如下：

for(int i =startIndex;i<=n;i++){//控制树的横向遍历
   path.push_back(i);//处理节点
   backtracking(n,k,i+1);//递归：控制树的纵向遍历，注意下一层搜索要从i+1开始
   path.pop_back();//回溯，撤销处理的节点
}

组合问题完整的C++代码如下：

class Solution{
private:
    vector<vector<int>> result;//存放符合条件结果的集合
    vector<int> path;//存放符合条件结果
    void backtracking(int n,int k,int 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();//回溯，撤销处理的节点
     }
  }
public:
    vector<vector<int>> combine(int n,int k){
         result.clear();//可以不写
         path.clear();//可以不写
         backtracking(n,k,1);
         return result;
  }
}