回溯算法之子集、组合、全排列

求全排列

思路

• track：记录一个全排列结果。
• 选择列表(nums)：选择一个满足条件的元素，加入到track中。

代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */

// 返回nums数组的全排列
var permute = function(nums) {
    let res = [];
    // 路径：记录在track中。
    // 选择列表: 从nums中选。
    // 结束条件：nums中所有数字都在track中出现。
    let fn = function(nums, track){
        if(track.length==nums.length){
            res.push([...track]);   // 坑1: track是数组,是引用类型,如果使用res.push(track),当track修改后, res也会变化。
            return;
        }

        // 从nums中选出不在track中的数字
        for(let i=0; i<nums.length; i++){
            if(track.indexOf(nums[i])==-1){
                track.push(nums[i]);
                fn(nums, track);
                track.pop();
            }
        }
    }
    fn(nums, []);
    return res;
};

求组合

思路

• track：记录一个有效的组合。
• 选择列表：要注意[1, 2]和[2, 1]不能重复出现。

代码

/**
 * @param {number} n
 * @param {number} k
 * @return {number[][]}
 */

// 回溯法
// 选择列表(for循环+if判断)
// 满足条件的某条路径(用数组存储)
var combine = function(n, k) {

    let res = [];
    let nums = [];
    for(let i=1; i<=n; i++){
        nums.push(i);
    }
    
    var fn = function(nums, k, track){
   		// 结束条件
        if(track.length==k){
            res.push([...track]);
        }
        for(let i=0; i<nums.length; i++){
            if(track.length==0 || nums[i]>track[track.length-1]){
            	// 做选择
                track.push(nums[i]);
                // 在该选择下，继续下一层递归，寻找下一个可选择的元素。
                fn(nums, k, track);
                // 递归完，重新选择此层元素。
                track.pop();
            }
        }
    }

    fn(nums, k, []);
    return res;
};

求子集

回溯思路

代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */

var subsets = function(nums) {
    // 回溯方法
    let res = []
    var back = function(nums, track){
    	// 结束条件
        if(track.length<=nums.length){
            res.push([...track]);
        }
        for(let i=0; i<nums.length; i++){
            if(track.length==0 || nums[i]>track[track.length-1]){
            	// 做选择
                track.push(nums[i]);
                // 做完选择，进入下一层递归继续选择
                back(nums, track);
                // 撤销当前层的选择。
                track.pop();
            }
        }
    }
    back(nums, []);
    return res;
};

递归方程思路

• subset([1, 2, 3]) = subset([1, 2]) + subset([1, 2])中每个列表元素加上3这个元素。

代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */

//  subset([1,2,3]) = subset([1, 2]) + (subset([1, 2])中每个元素加3
var subsets = function(nums) {
    if(nums.length==0) return [[]];
    let num = nums.pop()
    let subnums = subsets(nums);
    let res = [];
    for(let i=0; i<subnums.length; i++){
        res.push([...subnums[i]]);
        subnums[i].push(num);
        res.push([...subnums[i]]);
    }
    return res;
};

巩固练习

求有向无环图所有可能的路径