力扣77. 组合

最新推荐文章于 2024-09-09 18:27:53 发布

slowfastflow

最新推荐文章于 2024-09-09 18:27:53 发布

阅读量614

点赞数 9

分类专栏：力扣实践文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/N_BenBird/article/details/135613160

版权

力扣实践专栏收录该内容

148 篇文章 0 订阅

订阅专栏

本文介绍了如何使用递归和深度优先搜索（DFS）算法生成一个给定长度n的序列的所有可能k长度子序列。C++代码展示了如何在`Solution`类中实现这个过程，并处理了边界条件和结果存储。

摘要由CSDN通过智能技术生成

递归

思路：

找到一个长度为 nnn 的序列 aaa 的所有子序列，代码框架：

std::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);
}

本题递归结束条件为序列中元素为 k，即：

    void dfs(int cur, int n, int k) {
        // 序列元素个数满足条件，存储结果退出
        if (item.size() == k) {
            result.push_back(item);
            return;
        }

        // 考虑选择当前位置
        item.push_back(cur);
        dfs(cur + 1, n, k);
        item.pop_back();

        // 考虑不选择当前位置
        dfs(cur + 1, n, k);
    }

private:
    std::vector<int> item;
    std::vector<std::vector<int>> result;

已经构建的序列S'，剩余的数不足以构成满足条件的序列 S：
- 剩余的数长度 n - (cur - 1)
- size(S') + (n - (cur - 1)) < size(S) = k
- 这种情况也需要退出递归

综上，完整代码

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        dfs(1, n, k);
        return result;
    }

private:
    void dfs(int cur, int n, int k) {
        if (item.size() + (n -cur + 1) < k) {
            return;
        }

        if (item.size() == k) {
            result.push_back(item);
            return;
        }

        item.push_back(cur);
        dfs(cur + 1, n, k);
        item.pop_back();
        dfs(cur + 1, n, k);
    }

private:
    std::vector<int> item;
    std::vector<std::vector<int>> result;
};