[ 回溯 ] 组合总和III

最新推荐文章于 2025-12-26 16:25:00 发布

原创最新推荐文章于 2025-12-26 16:25:00 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

回溯专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个使用深度优先搜索（DFS）和回溯算法解决的组合总和问题变种——组合总和III。该问题要求从1到9中找出k个数使得其和为n，并通过剪枝优化算法提高效率。

216. 组合总和 III - 力扣（LeetCode） (leetcode-cn.com)

组合总和 III

DFS
回溯
剪枝，剪枝是根据条件剪枝，比如在for循环根据for剪枝，在边界处根据n剪枝

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> path;

    void DFS(int k, int n, int sum, int index) {
        // 剪枝
        if (sum > n) return;

        // 边界
        if (path.size() == k) {
            if (sum == n) ans.emplace_back(path);
            return;
        }
        
        for (int i = index; k - path.size() <= 9 - i + 1; ++i) {
            sum += i;
            path.emplace_back(i);
            DFS(k, n, sum, i + 1);
            
            // 回溯
            sum -= i;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        DFS(k, n, 0, 1);
        return ans;
    }
};