题目:
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
思路
从 n 中选择 k 个数
递归:
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
var combine = function (n, k) {
let _result = []
if (k === 0) return _result
function dfs(i, item) {
// n也参与组合,当指针越界则结束
if (i > n + 1 ) return
if (item.length === k) {
_result.push(item)
return
}
dfs(i + 1, [...item, i])
dfs(i + 1, item)
}
dfs(1, [])
return _result
}
超时
优化剪枝逻辑
即使指针没越界,指针之后元素全选也不能满足选择够k个元素则不用继续递归
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
var combine = function (n, k) {
let _result = []
if (k === 0) return _result
function dfs(i, item) {
// 当已选择的元素和未选择的元素数量和不能到达k个则递归结束
if (item.length + (n - i + 1) < k) return
if (item.length === k) {
_result.push(item)
return
}
dfs(i + 1, [...item, i])
dfs(i + 1, item)
}
dfs(1, [])
return _result
}