码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode之回溯法专题-39. 组合总数(Combination Sum)

时间:2019-08-07 22:11:30      阅读:62      评论:0      收藏:0      [点我收藏+]

标签:包括   solution   一个   数组   重复元素   sum   dex   remove   无限   

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

  • 所有数字(包括 target)都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例 2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]


分析:给定一个目标值,给定N个数字,要从N个数字里取数字(可以重复选取)
求这些数字的和等于目标值的所有情况。

用回溯法很好解决这个问题,首先从0-(n-1)中选取第1个数,第一个数取1-k次,以此类推。


AC代码如下:
class Solution {
    List<List<Integer>> ans = new ArrayList<>();
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        
        dfs(candidates,target,0,target,new ArrayList<Integer>());
        return ans;
    }
    
    
    public void dfs(int[] candidates,int target,int index,int balance,ArrayList<Integer> list){
        if(balance<0){
            return;
        }
        if(balance==0){

            ans.add(new ArrayList<Integer>(list));
            return;
        }
        
        if(index>=candidates.length){
            return;
        }
        
        int num = candidates[index];
        int n = balance/num +1;
        

        for(int i=0;i<=n;i++){
            
            for(int j=0;j<i;j++){
                list.add(num);
            }
            int count = num*i;
            dfs(candidates,target,index+1,balance-count,list);
            for(int j=0;j<i;j++){
                list.remove(list.size()-1);
            }
        }
        
    }
}

 




Leetcode之回溯法专题-39. 组合总数(Combination Sum)

标签:包括   solution   一个   数组   重复元素   sum   dex   remove   无限   

原文地址:https://www.cnblogs.com/qinyuguan/p/11318109.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!