LeetCode_216(回溯)

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

说明:

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

示例 1:

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

示例 2:

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

        本题用到回溯,首先定义一个变量sum代表当前搜索到的数字序列的总和,再定义一个pos变量代表当前遍历的层数,即代表当前取了几个数,从sum=0,pos=0开始递归搜索。每一层都遍历1到9,从中取一个数,sum加上当前取的数,pos也+1,然后递归到下一层,每递归一层就加上当前层遍历到的那个数,同时层数+1。当pos=k时判断sum是否等于n,若等于则将数字序列加入解空间中。若还没有递归到第k层但是总和已经超过了n则返回,若层数超过了k也返回,即回溯操作。代码如下:

public List<List<Integer>> combinationSum3(int k, int n) {
		List<List<Integer>> res = new ArrayList<>();
		List<Integer> temp = new ArrayList<>();
		dfs(res, temp, 0, n, k, 0);
		return res;
	}

	void dfs(List<List<Integer>> res, List<Integer> temp, int pos, int n, int k, int sum) {
		if (pos < k && sum > n)
			return;
		if (pos == k && sum == n) {
			res.add(new ArrayList<Integer>(temp));
			return;
		}
		if (pos > k)
			return;
		for (int i = 1; i <= 9; i++) {
			if (temp.contains(i))
				return;
			temp.add(i);
			sum += i;
			dfs(res, temp, pos + 1, n, k, sum);
			temp.remove(temp.size() - 1);
			sum -= i;
		}
	}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值