一天一大 lee(组合)难度:中等-Day20200908
题目:
给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
抛砖引玉
思路
从 n 中选择 k 个数
- 对于任意一个数存在选择和不选择两种情况，其中这两种情况有分别对应着不同的组合
递归：
- 参数：
- 枚举分支的指针
- 枚举分支的中间组合数组
- 终止：
- 组合数组元素数量等于 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
}举报
腾讯云开发者
