forked from walnutown/CodingInTheDeep
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCombinationSum2.java
More file actions
71 lines (64 loc) · 2.68 KB
/
Copy pathCombinationSum2.java
File metadata and controls
71 lines (64 loc) · 2.68 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
Given a collection of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.
Each number in C may only be used once in the combination.
Note:
All numbers (including target) will be positive integers.
Elements in a combination (a1, a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
The solution set must not contain duplicate combinations.
For example, given candidate set 10,1,2,7,6,1,5 and target 8,
A solution set is:
[1, 7]
[1, 2, 5]
[2, 6]
[1, 1, 6]
*/
// This is known as SubsetSum problem, a NP-complete problem
// http://en.wikipedia.org/wiki/Subset_sum_problem
// Backtracking
// The challenge mainly lies in how to avoid duplicates
// candidates are not unique ("Given a collection of candidate numbers")
// time: O(2^n) why? There're 2^n subsets, we almost check each one.
public class Solution {
public ArrayList<ArrayList<Integer>> combinationSum2(int[] num, int target) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
if (num == null || num.length == 0) return res;
Arrays.sort(num);
finder(num, 0, target, res, new ArrayList<Integer>());
return res;
}
public void finder(int[] num, int index, int target, ArrayList<ArrayList<Integer>> res, ArrayList<Integer> r){
if (target < 0) return;
if (target == 0){
res.add(new ArrayList<Integer>(r));
return;
}
for (int i = index; i< num.length; i++){
if (i > index && num[i] == num[i-1]) continue; // avoid duplicates in result
r.add(num[i]);
finder(num, i+1, target-num[i], res, r);
r.remove(r.size()-1);
}
}
}
// DFS, use set to avoid duplicates, worse performance than the above
public class Solution {
public ArrayList<ArrayList<Integer>> combinationSum2(int[] num, int target) {
Set<ArrayList<Integer>> res = new HashSet<ArrayList<Integer>>();
if (num == null || num.length == 0) return new ArrayList<ArrayList<Integer>>(res);
Arrays.sort(num);
finder(num, 0, target, res, new ArrayList<Integer>());
return new ArrayList<ArrayList<Integer>>(res);
}
public void finder(int[] num, int index, int target, Set<ArrayList<Integer>> res, ArrayList<Integer> r){
if (target < 0) return;
if (target == 0){
res.add(new ArrayList<Integer>(r));
return;
}
for (int i = index; i< num.length; i++){
r.add(num[i]);
finder(num, i+1, target-num[i], res, r);
r.remove(r.size()-1);
}
}
}