216. Combination Sum III

最新推荐文章于 2023-05-22 12:05:46 发布

zibingling777

最新推荐文章于 2023-05-22 12:05:46 发布

阅读量81

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题笔记文章标签： Backtracking Array

本文链接：https://blog.youkuaiyun.com/zibingling777/article/details/78811992

刷题笔记专栏收录该内容

304 篇文章

订阅专栏

本文介绍了一种使用回溯法解决从1-9中选取k个数使其和为n的问题，并提供了详细的代码实现。通过递归的方式进行深度优先搜索，确保组合中元素的数量为k且总和等于n。

1、题目描述

输入两个整数k和n，从1-9中选k个数，使得k数之和为n，返回所有这样的k元组。

2、思路

Backtracking。

和II差不多，比II简单的一点是，数组不会有重复元素，所以dfs的时候不用检查了。

另外需要一个变量来记录组成小组合的元素数量，确保是k个数，和为n。

3、代码

vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> ans;
        vector<int>v;
        dfs(ans,v,n,k,1);
        return ans;
    }
    void dfs(vector<vector<int>>& ans,vector<int>& v, int n, int k, int start){
        if(n==0){
            if(k==0){
                ans.push_back(v);
                return;
            }
            else
                return;
        }
        
        else if(n<0||k==0)
            return;
        
        for(int i=start;i<=9;i++){            
                v.push_back(i);
                dfs(ans,v,n-i,k-1,i+1);
                v.pop_back();                
        }
    }