LeetCode-----第七十七题-----组合

组合

难度：中等

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

题目解析：

       显然使用回溯法，仔细分析一下。先建立一个1~n的数组，我们需要一个全局index去作为for循环的初始值。这还有重复问题。

       结束条件：当计数器计到和所要抽取的数字个数相等时结束，将temp压入res

       all station：因为题目不要求重复，所以传入给index的值为i+1（下一个数字，这样也完成了剪枝，如果是index+1传入的话，会发生重复），并同时计数+1

 

参考代码：

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> res;
        if(n < 1 || k < 1 || k > n)
            return res;
        
        vector<int> temp;
        vector<int> nums;
        nums.resize(n);
        for(int i = 0; i < n; i++)
        {
            nums[i] = i+1;
        }
        back_search(res,nums,temp,0,n,k,0);
        return res;
    }
    void back_search(vector<vector<int>>& res, vector<int>& nums, vector<int>& temp, int index, int n,int k,int count)
    {
        //结束条件，当计数器和所要提取的数目相等时结束
        if(count >= k)
        {
            if(count == k)
                res.push_back(temp);
            return;
        }

        //all station
        for(int i = index; i < n; ++i)
        {
            temp.push_back(nums[i]);
            back_search(res,nums,temp,i+1,n,k,count+1);//将下一个数字传给下一次迭代，而不是index+1
            temp.pop_back();
        }
    }
};