LeetCode78 子集(带扩展)

leetcode.cn/problems/su...

解法一:回溯

因为集合中的元素不用考虑顺序,我们通过保证元素之间的相对顺序不变来防止出现重复的子集

例如,[1,2,3]2 后面只有 3,如果你添加了前面的 1,那么后面就只能选择23,若选择 了2,后面就只能选择3,否则,如果出现[2,1],就会和之前已经生成的子集 [1,2] 重复了。

得到下面这棵决策树

这棵树的特性是:如果把根节点作为第 0 层,那么第 n 层的所有节点就是大小为 n 的所有子集

题意要求得到所有子集,那只要遍历这棵多叉树,把所有节点的值加入结果即可

go 复制代码
func subsets(nums []int) [][]int {
	path := make([]int, 0)
	var res [][]int
	backtrack(nums, 0, path, &res)
	return res
}

func backtrack(nums []int, offset int, path []int, res *[][]int) {
	if len(path) <= len(nums) { // 每个节点的值都是一个子集
		tmp := make([]int, len(path))
		copy(tmp, path)
		*res = append(*res, tmp)
	}
	for i := offset; i < len(nums); i++ { // 递归退出条件: offset == len(nums)
		path = append(path, nums[i]) // 做选择
		backtrack(nums, i+1, path, res) // 通过offset参数控制树枝的遍历,每次只能往后选择元素,避免产生重复的子集
		path = path[:len(path)-1] // 撤销选择
	}
}

扩展

组合问题和子集问题其实是等价的 ,给定一个数组,大小为 k 的组合就是大小为 k 的子集。

来看这道题:leetcode.cn/problems/co...

回想刚才的决策树,其实这个题就是找出第k层的所有节点,前面的子集代码稍作改动即可

go 复制代码
func combine(n int, k int) [][]int {
    nums := make([]int, 0, n)
    for i:=1; i<=n; i++{
        nums = append(nums, i)
    }
    path := make([]int, 0)
    var res [][]int 
    backtrack(nums, 0, path, k, &res)
    return res
}

func backtrack(nums []int, offset int, path []int, k int, res *[][]int){
    if len(path) == k{ // 收集长度为k的子集
        tmp := make([]int, len(path))
        copy(tmp, path)
        *res = append(*res, tmp)
        return
    }
    for i:= offset; i<len(nums); i++{ // 通过offset避免重复选择元素
        path = append(path, nums[i]) // 做选择
        backtrack(nums, i+1, path, k, res)
        path = path[:len(path)-1] // 撤销选择
    }
}

可以再优化一下空间复杂度,不需要生成一个额外的数组nums

go 复制代码
func combine(n int, k int) [][]int {
    path := make([]int, 0)
    var res [][]int
    backtrack(n, 1, path,k, &res) // 范围是【1,n】
    return res
}

func backtrack(n int, offset int, path []int, k int, res *[][]int) {
    if len(path) == k{
        *res = append(*res, append([]int{}, path...))
        return
    }
    for i:=offset; i<=n; i++{
        path = append(path, i)
        backtrack(n, i+1, path, k, res)
        path = path[:len(path)-1]
    }
}
相关推荐
拾光拾趣录1 分钟前
单点登录(SSO):实现跨系统无缝访问
前端·后端
知其然亦知其所以然2 分钟前
对象访问两种方式PK:句柄 VS 指针,谁才是JVM最爱?
java·后端·面试
AKAMAI11 分钟前
在Akamai平台上进行VOD转码的参考架构
后端·云原生·云计算
Dream耀15 分钟前
从零实现JavaScript防抖与节流:渐进式优化之旅
前端·javascript·面试
前端布鲁伊17 分钟前
【前端面试场景题】JavaScript数字溢出了?BigInt来救场!前端处理超大数值的完整攻略
前端·面试
代码与野兽17 分钟前
Web3开发者生态终极选择指南:新手如何在激烈竞争中脱颖而出?
前端·后端·web3
sohoAPI18 分钟前
Flask快速入门
后端·python·flask
hai99long22 分钟前
利用Atomikos实现XA分布式事务
后端
花落人散处27 分钟前
SpringAI —— 接入DeepSeek
java·后端
Jacob023435 分钟前
“Node.js 不行了”?性能争议中的误解与选择真相
后端·node.js