LeetCode78 子集(带扩展)

11 阅读3分钟

leetcode.cn/problems/su…

image.png

解法一:回溯

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

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

得到下面这棵决策树

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

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

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] // 撤销选择
	}
}

扩展1: 组合问题

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

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

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

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

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]
    }
}

扩展2:子集Ⅱ (剪枝策略)

前面讲的标准子集/组合问题输入的 nums 都是没有重复元素的,但如果存在重复元素,又该怎么处理?

来看这道题 leetcode.cn/problems/su…

image.png

概括一下题意就是求元素可重复但不可复选的所有子集 以 nums = [1,2,2] 为例,为了区别两个 2 是不同元素,后面我们写作 nums = [1,2,2']。同样的,我们画出这棵决策树 image.png 显然,2和2‘两条值相同的相邻树枝会产生重复,需要进行剪枝,如果一个节点有多条值相同的树枝相邻,则只遍历第一条,剩下的都剪掉 image.png

func subsetsWithDup(nums []int) [][]int {
    // 先排序,让相同的元素靠在一起
    sort.Slice(nums, func (i, j int) bool{
        return nums[i] <= nums[j]
    })
    path := make([]int, 0)
    var res [][]int
    backtrack(nums, 0, path, &res)
    return res
}

func backtrack(nums []int, offset int, path []int, res *[][]int) {
    // 前序位置,每个节点的值都是一个子集
    tmp := make([]int, len(path))
    copy(tmp, path)
    *res = append(*res, tmp)
    for i := offset; i<len(nums); i++{
        if i > offset && nums[i] == nums[i-1]{ // 剪枝
            continue
        }
        path = append(path, nums[i])
        backtrack(nums, i+1, path, res)
        path = path[:len(path)-1]
    }
}