Leetcode:组合

回溯法代码框架

vector<int> temp;
void dfs(int cur, int n) {
    if (cur == n + 1) {
        // 记录答案
        // ...
        return;
    }
    // 考虑选择当前位置
    temp.push_back(cur);
    dfs(cur + 1, n, k);
    temp.pop_back();
    // 考虑不选择当前位置
    dfs(cur + 1, n, k);
}

回溯法与二叉树的深度优先遍历异曲同工

class Solution {
public:
	void backtrack(vector<vector<int>> &result,vector<int> &ans,int num,int n,int k){
		if(ans.size()+(n-num+1)<k){
			return;
		}
		if(ans.size()==k){
			result.push_back(ans);
			return;
		}
		ans.push_back(num);
		//子结果的控制因素,如确定vector的起始元素是1、2,在后面变化元素防止重复 
		backtrack(result,ans,num+1,n,k);
		ans.pop_back();
		//在把末尾的元素弹出后,需要继续加入新元素,所以这里要加一个回溯 
		backtrack(result,ans,num+1,n,k);
	}
    vector<vector<int>> combine(int n, int k) {
		vector<vector<int>> result;
		vector<int> ans;
		backtrack(result,ans,1,n,k);
		return result;
    }
};
posted @ 2021-12-26 15:37  Dreamer_szy  阅读(31)  评论(0)    收藏  举报