LeetCode Combination Sum III

原创 2015年07月09日 16:42:16

Description:

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.

Ensure that numbers within the set are sorted in ascending order.

Solution:

这种DFS有两种写法,这里进行总结。

写法一:

每层的递归是根据当前这个数字是否添加进行。所以最后终止条件有两个

1.递归到10

2.当前的ArrayList中包含的数字正好有k个

复杂度为2^10


写法二:

每层的递归是根据第i个选取的数字是多少。递归地终止条件只要一个,就是加入ArrayList的数字有k个。

复杂度为2^k


写法一代码:

import java.util.*;

public class Solution {
	int n, k;
	List<List<Integer>> list = new ArrayList<List<Integer>>();

	public List<List<Integer>> combinationSum3(int k, int n) {
		this.k = k;
		this.n = n;

		dfs(1, 0, new ArrayList<Integer>());
		return list;
	}

	void dfs(int num, int tempSum, ArrayList<Integer> array) {
		if (num == 10) {
			if (tempSum == n && array.size() == k)
				list.add(new ArrayList<Integer>(array));
			return;
		}

		dfs(num + 1, tempSum, array);
		if (tempSum + num <= n) {
			array.add(num);
			dfs(num + 1, tempSum + num, array);
			array.remove(array.size() - 1);
		}
	}
}
写法二代码:

很明显写法二比较适合这道题目,需要注意的就是每一层,选取第selectedNum+1个元素的时候,元素的范围不是无限制的,需要小于等于 10 - ( k - selectedNum);如果直接是小于10也可以,那就需要递归的时候判断一下,num<10的条件是否成立,否则num会等于10

import java.util.*;

public class Solution {
	int n, k;
	List<List<Integer>> list = new ArrayList<List<Integer>>();

	public List<List<Integer>> combinationSum3(int k, int n) {
		this.k = k;
		this.n = n;

		dfs(1, 0, 0, new ArrayList<Integer>());
		return list;
	}

	void dfs(int num, int selectedNum, int tempSum, ArrayList<Integer> array) {
		if (selectedNum == k) {
			if (tempSum == n)
				list.add(new ArrayList<Integer>(array));
			return;
		}

		for (int i = num; i <= 10 - (k - selectedNum); i++)
			if (tempSum + i <= n) {
				array.add(i);
				dfs(i + 1, selectedNum + 1, tempSum + i, array);
				array.remove(array.size() - 1);
			}

	}

}


版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

Leetcode|Combination Sum III[回溯]

Find all possible combinations of k numbers that add up to a number n, given that only numbers from ...

leetcode 216. Combination Sum III

class Solution { void do_once(int k,int n, map, int>&candi) { map, int>newcandi; for (map, int...

LeetCode #216 - Combination Sum III - Medium

Combination Sum Series Combination Sum I: Combination Sum II: Combination Sum III: Pr...

leetcode-java-216. Combination Sum III

/* Given a set of candidate numbers (C) and a target number (T), find all unique combinations in C w...

<LeetCode OJ> 39 / 40 / 216 Combination Sum(I / II / III)

Total Accepted: 83289 Total Submissions: 274461 Difficulty: Medium Given a set of candidate num...

leetcode -- Combination Sum III -- 重点,dfs回溯模板

https://leetcode.com/problems/combination-sum-iii/简单。总结自己的回溯模板class Solution(object): def dfs(sel...

LeetCode216:Combination Sum III

从1-9中取k个数和为n,且数不重复。首先可能想到的是用一个boolean数组记录用过的数字,Boolean数组的下标+1即为数字。但这题我们发现,选取的数组是递增的,所以无需使用一个数组记录使用过的...

leetcode:Combination Sum III

Find all possible combinations of k numbers that add up to a number n, given that only numbers fro...

Leetcode 39 - Combination Sum I, II, III(dfs)

题意给一个set C,然后还有一个你要凑出来的数target,要求C里面的数可以使用任意多次,求方案(输出具体方案)。思路先将数组从大到小排序(cut),然后开始搜索。 注意dfs模型:void d...
  • Lzedo
  • Lzedo
  • 2017-01-24 17:29
  • 199

【LeetCode-面试算法经典-Java实现】【216-Combination Sum III (组合数的和)】

【216-Combination Sum III (组合数的和)】【LeetCode-面试算法经典-Java实现】【所有题目目录索引】代码下载【https://github.com/Wang-Jun-...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)