https://oj.leetcode.com/problems/combinations/
Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.
For example,
If n = 4 and k = 2, a solution is:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
public List<List<Integer>> combine(int n, int k)
和Combinations Sum 以及 Combinations Sum II是一类型的题目。建议大家也熟悉一下下一题Subset的做法,那是一切Combination类型题目的鼻祖,所有同类题目都是由那题延伸而来。另外还要注意区分一下Permutation和Subset的做法就好。这一题和最原始类型的combination的做法区别就在于这一题只需要且仅需要当长度为k的时候录入buf里的结果,其余都是不用的。
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> res = new LinkedList<List<Integer>>();
if(k > n)
return res;
Integer[] buf = new Integer[k];
helper(res, buf, 0, 0, n, k);
return res;
}
public void helper(List<List<Integer>> res, Integer[] buf, int curpos, int curlevel, int n, int k){
if(curpos == k){
Integer[] cur_res = new Integer[k];//
System.arraycopy(buf, 0, cur_res, 0, k);
res.add(Arrays.asList(cur_res));//建议大家注意一下这里的写法,之所以要做拷贝再录入是因为asList这个函数会直接把当前数组作为list的内容
//也就是asList里面的实际数组内容就是buf这个reference。所以不能直接用buf
}else if(k - curpos > n - curlevel){
return;
}else{
for(int i = curlevel; i < n; i++){
buf[curpos] = i + 1;
helper(res, buf, curpos + 1, i + 1, n, k);
}
}
}