C++ 递归 / 溯回

LK.77 组合：

给定两个整数 n 和 k,返回范围[1, n]中所有可能的 k 个数的组合。

示例 1：输入：n = 4, k = 2

              输出：[ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]

示例2：输入：n = 1, k = 1

             输出：[ [1] ]

vector<int> temp;
vector<vector<int>> ans;

void dfs(int cur, int n, int k) {
    // 剪枝：temp 长度加上区间 [cur, n] 的长度小于 k，不可能构造出长度为 k 的 temp
    if (temp.size() + (n - cur + 1) < k) {
        return;
    }
    // 记录合法的答案
    if (temp.size() == k) {
        ans.push_back(temp);
        return;
    }
    // 考虑选择当前位置
    temp.push_back(cur);
    dfs(cur + 1, n, k);
    temp.pop_back();
    // 考虑不选择当前位置
    dfs(cur + 1, n, k);
}

vector<vector<int>> combine(int n, int k) {
    dfs(1, n, k);
    return ans;
}

LK.78 子集：

给一个整数数组 nums ，数组中的元素互不相同 。返回该数组所有可能的子集（幂集）。

eg: 输入：nums = [1,2,3]  输出：[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

解题：

（1）迭代法

vector<vector<int>> subsets(vector<int>& nums) {
    vector<vector<int>> ans;
    vector<int> temp;
    int len = nums.size();
    for (int mask = 0; mask < 1 << len; mask++) {
        temp.clear();
        for (int i = 0; i < len; i++) {
            if (mask & 1 << i) {
                temp.push_back(nums[i]);
            }
        }
        ans.push_back(temp);
    }    
    return ans;
}

（2）递归

vector<vector<int>> ans;
vector<int> temp;
int len;

void dfs(vector<int>& nums, int k, int x) {
    if (temp.size() == k) {
        ans.push_back(temp);
        return;
    }
    if (x >= len)
        return;
    
    temp.push_back(nums[x]);
    dfs(nums, k, x + 1);
    temp.pop_back();
    dfs(nums, k, x + 1);
}

vector<vector<int>> subsets1(vector<int>& nums) {
    len = nums.size();
    ans.push_back(temp);

    for (int i = 0; i < len; i++) {
        dfs(nums, i + 1, 0);
    }
    return ans;
}