剑指 Offer II 080. 含有 k 个元素的组合(77. 组合)
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
提示:
1 <= n <= 20
1 <= k <= n
通过次数292,873提交次数
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
参考:https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0077.%E7%BB%84%E5%90%88.md
题解(Java)
三部曲法:
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> ans = new LinkedList<>();
helper(ans, new LinkedList<Integer>(), 1, n, k);
return ans;
}
private void helper(List<List<Integer>> ans, LinkedList<Integer> record, int startIndex, int n, int k) {
if (record.size() == k) {
ans.add(new LinkedList<>(record));
return;
}
for (int i = startIndex; i <= n - (k - record.size()) + 1; i++) {
record.add(i);
helper(ans, record, i + 1, n, k);
record.removeLast();
}
}
}
选/不选法:
class Solution {
List<List<Integer>> ans = new LinkedList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> combine(int n, int k) {
helper(n, k, 1);
return ans;
}
private void helper(int n, int k, int start) {
if (path.size() == k) {
ans.add(new LinkedList<Integer>(path));
} else if (start <= n && start <= n - (k - path.size()) + 1) {
helper(n, k, start + 1);
path.add(start);
helper(n, k, start + 1);
path.removeLast();
}
}
}