给定两个整数n
和k
,返回1 ... n
中所有可能的k
个数的组合。
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
/**
* @param {number} n
* @param {number} k
* @return {number[][]}
*/
var combine = function(n, k) {
if(n <= k) return [[...new Array(n).keys()].map(v => v+1)];
var target = [];
var dfs = function(cur, deep, tmp){
if (tmp.length + (n - cur + 1) < k) return void 0;
if(deep === k){
target.push(tmp);
return void 0;
}
for(;cur<=n;++cur) dfs(cur+1, deep+1, [...tmp, cur]);
}
dfs(1, 0, []);
return target;
};
以示例中的值为例,可以认为是一个长度为4
的数组[1, 2, 3, 4]
,每两个组合一个数组可取1
组合其数组中之后的值,2
与其数组中之后值,3
与其数组中之后的值,4
与其数组中之后值,即[1, 2]
、[1, 3]
、[1, 4]
、[2, 3]
、[2, 4]
、[3, 4]
,首先初始条件判断,若是n <= k
则只能构成一个长度为n
的数组,将其装入二维数组返回即可,后边的表达式利用了new Array(n)
生成了一个长度为n
的空数组,让后取得其keys
的迭代器,利用...
即Spread
操作符将其展开,之后使用map
将其处理为key
值+1
,之后定义目标数组,之后定义dfs
递归函数,首先进行剪枝,如果当前tmp
数组的大小为s
,未确定状态的区间[cur,n]
的长度为t
,如果s + t < k
,那么即使t
个都被选中,也不可能构造出一个长度为k
的序列,故这种情况就没有必要继续向下递归,之后判断递归深度如果与k
相等则直接将tmp
数组置入目标数组并返回,之后定义一个循环,从cur
开始到n
进行递归取值,将tmp
数组与cur
构建一个新数组传递到下一个递归中,之后启动递归初始化cur
为1
,深度deep
为0
,tmp
为一个空数组,递归完成后返回目标数组即可。
https://github.com/WindrunnerMax/EveryDay
https://leetcode-cn.com/problems/combinations/