leetcode-组合:回溯+剪枝

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

第一次这么快的解出一个中等题,不过这个本身也很简单,但是要想到如何剪枝才能提高效率:
比如n=4,k=3的情况,{1,2,3,4}
要是递归到pos=3的情况,那根本不可能最后总数可以到3,这种情况应当被剪去

class Solution {
     
public:
    vector<vector<int>> combine(int n, int k) {
     
        vector<vector<int>> res;
        vector<int> arr;
        recruise(arr,res,1,n,k);
        return res;
    }
    void recruise(vector<int> &arr,vector<vector<int>> &res,int pos,int n,int k)
    {
     
        if(arr.size()==k)
        {
     
            res.push_back(arr);
            return;
        }
        if(n-pos+1<k-arr.size())
            return;
        //若剩下的可插入数都无法满足达到最后要求的数,那直接退出
        
        for(int i=pos;i<=n;i++)
        {
     
            arr.push_back(i);
            recruise(arr,res,++pos,n,k);
            arr.pop_back();
        }

    }
};

你可能感兴趣的:(leetcode)