题目链接:https://leetcode.com/problems/combinations/description/
描述:
Given two integers n and k, return all possible combinations of k numbers out of 1 … n.
Example:
Input: n = 4, k = 2
Output:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
就是说在n个数中选出2个数,一共可能的排列。
此题可以归结为选或者不选的一类问题,其实代码结构也比较固定(用下边的程序进行说明):
- 新建一个保存最终答案的list(
ans
) 新建一个保存中间结果的list(list
) - 新建一个函数,并传递当前遍历到了树的第几层,也就是下面代码里边的level 每次list先选中当前元素(
list.add()
) - 递归下一层
- 之后list不选当前的元素(
list.remove())
- 递归下一层
- 递归的结束条件根据题意来定,但是记得最后要新拷贝一份list保存在答案中,而不是直接将答案保存,(是
ans.add(new ArrayList<Integer>(list))
;而不是ans.add(list)
)。这是因为,java中一切对象皆引用,如果用后者,最后ans中保存的将是一模一样的n个list。
class Solution {
public List<List<Integer>> combine(int n, int k) {
assert n >= 1 && k <= n;
List<List<Integer>> ans = new ArrayList<>();
List<Integer> list = new ArrayList<>();
dfs(1, n, k, ans, list);
return ans;
}
private void dfs(int level, int n, int k, List<List<Integer>> ans, List<Integer> list) {
if(list.size() == k) {
ans.add(new ArrayList<Integer>(list));
return;
}
if(level > n) {
return;
}
// choose this element
list.add(level);
dfs(level+1, n, k, ans, list);
// dosen't choose this element
list.remove(list.size() - 1);
dfs(level+1, n, k, ans, list);
// 就像画全排列树一样,这个元素是选还是不选,靠着代码就能说清楚
}
}