leetcode39. 组合总和（回溯+剪枝）-小羊的记录本

leetcode39. 组合总和

给定一个无重复元素的数组 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]
]

提示：

1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500

思路：
1.回溯
2.剪枝，因为最后结果不能重复

图来源于liweiwei
代码

class Solution {
    public List<List<Integer>> combinationSum(int[] candidates, int target){
    	//要排序，及时终止，避免无用功
        Arrays.sort(candidates);
    	List<List<Integer>> res = new ArrayList<>();
    	List<Integer> tmp = new ArrayList<Integer>();
    	
    	dfs(candidates, target, 0, tmp, res);  	  	
    	return res;
    }
    //begin很重要，不然就会有重复的结果
    //target是每次要寻找的下一个目标
    public void dfs(int[] candidates, int target,int begin, List<Integer> tmp, List<List<Integer>> res) {
		//target==0才是成功的
		if(target == 0) {
			res.add(new ArrayList<Integer>(tmp));
			return;
		}
		
		for(int i = begin; i < candidates.length; i++) {
			//一旦发现剩余的数过大，直接break，无需再找（后面的更大）
			if(candidates[i] > target)
				break;
			//add
			tmp.add(candidates[i]);
			//dfs 继续找下一目标，要从i开始，for循环内i不变，保证可重复用一个数；for循环外i++，保证不会用到前面的数，避免重复
			dfs(candidates, target - candidates[i], i, tmp, res);
			//remove
			tmp.remove(tmp.size()-1);
		}
    }
}