LeetCode每日一题 2020-9-11 216.组合总和III 回溯dfs 子集树 +剪枝

LeetCode每日一题 2020-9-11 216.组合总和III 回溯dfs+剪枝

题目描述
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:
所有数字都是正整数。
解集不能包含重复的组合。

示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii

解题思路
本题可以使用回溯子集树的方法来解决,子集树相当于对于每一个数选择加入或不加入。同时在搜索的过程中可以使用剪枝函数。最后需要注意的就是选择数字的个数,需要等于k。

题解代码

class Solution {
public:
   int leftSum[11]={0};
    void dfs(int t,int n,int k,int sum,vector<vector<int>>&ans,vector<int>& temp){
        if(sum==n){
            if(temp.size()==k){//正好是k个数
                ans.push_back(temp);
            }
            return;
        }
        if(sum+leftSum[t]<n){//剪枝
            return;
        }
        if(t>9){
            return;
        }
        temp.push_back(t);
        sum+=t;
        dfs(t+1,n,k,sum,ans,temp);
        temp.pop_back();
        sum-=t;
        dfs(t+1,n,k,sum,ans,temp);
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        for(int i=9;i>0;i--){
            leftSum[i]+=(i+leftSum[i+1]);
        }
        vector<vector<int>> ans;
        vector<int> temp;
        dfs(1,n,k,0,ans,temp);
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值