java实现回溯算法实例

java实现回溯算法实例

题目：216.组合总和 III

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

说明：

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

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

结题思路：

采用回溯的方法，根据示例1，查找分为3（k）层，第一层遍历所有元素，第二层在第一层基础上加一遍历，第三层在第二层基础上加一遍历。
结束条件：当查找求和大于等于预定值时，或者全部遍历后，或者已经遍历三层，结束。并将恰好等于的数字进行记录。
否则，记录当前值，并且继续调用本身。
当遍历完第三层，减去末尾值进行回溯。

代码

public class Sloution {
    public List<List<Integer>> combinationSum3(int k, int n) {
    	List<List<Integer>> lists = new ArrayList<List<Integer>>();
    	List<Integer> list = new ArrayList<Integer>();
    	int sum = 0;
    	
    	dfs(1, k, n, lists, list, sum);
    	return lists;
    }
    
    private void dfs(int index, int k, int n, List<List<Integer>> lists, List<Integer> list, int sum) {
        if(sum >= n || index >= n || list.size() >= k){	//终止条件
        	if(sum == n && list.size() == k){
            	lists.add(new ArrayList<>(list));
            }
        	return;
        }
    	else {	
    		for(int i=index; i<n; i++){
    			list.add(i);
                sum += i;
                dfs(++index, k, n, lists, list, sum);
                sum -= list.get(list.size()-1);
                list.remove(list.size()-1);
    		}
		}
        
    }
    
}