组合

前端之家收集整理的这篇文章主要介绍了组合前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

组合

给定两个整数nk,返回1 ... n中所有可能的k个数的组合。

示例

输入: n = 4,k = 2
输出:
[
  [2,4],[3,[2,3],[1,2],]

题解

/**
 * @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,[]);
    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构建一个新数组传递到下一个递归中,之后启动递归初始化cur1,深度deep0tmp为一个空数组,递归完成后返回目标数组即可。

每日一题

https://github.com/WindrunnerMax/EveryDay

参考

https://leetcode-cn.com/problems/combinations/

猜你在找的程序笔记相关文章