LeetCode 216. 组合总和 III java

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

说明:

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

示例 1:

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

1
2

示例 2:

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

1
2

解析:
相比于前面两道用回溯算法解题的题目:分割回文串和复原IP地址,这道题显得简单一些,并且也是更加典型的回溯解法。可以结合三道题目来看,解题方法很类似,加深感觉。

题目要求所有的和为 n 的 k 个数的组合,组合中只允许出现1-9的数字。从回溯算法的解题思路来出发,我们要走k步能够得到最后的符合条件的结果,这是递归的深度,所以需要在递归的参数中传递这个值,并且每次减小1,直到减小到0,然后看看这里得到的结果是不是符合条件的结果,如果是,那么就加入最终的解的集合,如果不是,那么就回退继续寻找。看是否符合条件是看前面步骤经过的数字之和是否为n。

另外,我们知道回溯算法每一步都是由多个分叉口的,也就是可以有多个选择,我们需要遍历每一种路径。题目要求每一个解不能有重复数字,所以我们按照从左至右选择的方式,每一步选择数组的范围都是从上一步选择位置的后面开始的。也就是代码中for循环的初始条件。

package nunber2;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Scanner;

public class Demo1 {
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		System.out.println("请输入k和n:");
		int k=sc.nextInt();
		int n=sc.nextInt();
		
		List<List<Integer>>  l=combinationSum3(k,n);
		
//		Iterator it= l.iterator();
//		Object list=new ArrayList<>();
//		while (it.hasNext()) {
//			list=it.next();
//			System.out.println(list);
//		}
		
		for (List<Integer> list : l) {
			System.out.println(list);
		}
	}
	 public static List<List<Integer>> combinationSum3(int k, int n) {
	        List<List<Integer>> res = new ArrayList<>();
	        ArrayList<Integer> templist = new ArrayList<>();
	        combinationSum(res, templist, k, n, 1);
	        return res;
	    }

	    /**
	     * 回溯算法
	     *
	     * @param res      结果集
	     * @param templist 临时存放list
	     * @param k        还需要几步递归结束
	     * @param n        对于目前还有k步的情况下,需要得到的和为n
	     * @param from     为了保证每个组合中都不存在重复,需要从from开始选择数字
	     */
	    private static void combinationSum(List<List<Integer>> res, ArrayList<Integer> templist, int k, int n, int from) {
	        if (n < 0) {//如果是一个负数,就要舍掉
	            return;
	        }
	        //走到这一步一定需要return
	        if (k == 0) {
	            if (n == 0) {
	                res.add(new ArrayList<>(templist));//把得到的结果加进去
	            }
	            return;
	        }
	        //为了保证每个组合中都不存在重复,需要从from开始选择数字,最后结束选择的范围是9 - k + 1
	        for (int i = from; i <= 9 - k + 1; i++) {
	            templist.add(i);
	            combinationSum(res, templist, k - 1, n - i, i + 1);
	            templist.remove(templist.size() - 1);//回溯算法常规步骤
	        }
	    }
}//

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值