📑 题目:90. 子集 II

🚀 本题 LeetCode 传送门

题目大意

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。说明:解集不能包含重复的子集。

解题思路

  • 这一题是第 78 题的加强版,比第 78 题多了一个条件,数组中的数字会出现重复。
  • 解题方法依旧是 DFS,需要在回溯的过程中加上一些判断。
  • 这一题和第 78 题,第 491 题类似,可以一起解答和复习。

代码

  1. package leetcode
  2. import (
  3. ""fmt""
  4. ""sort""
  5. )
  6. func subsetsWithDup(nums []int) [][]int {
  7. c, res := []int{}, [][]int{}
  8. sort.Ints(nums) // 这里是去重的关键逻辑
  9. for k := 0; k <= len(nums); k++ {
  10. generateSubsetsWithDup(nums, k, 0, c, &res)
  11. }
  12. return res
  13. }
  14. func generateSubsetsWithDup(nums []int, k, start int, c []int, res *[][]int) {
  15. if len(c) == k {
  16. b := make([]int, len(c))
  17. copy(b, c)
  18. *res = append(*res, b)
  19. return
  20. }
  21. // i will at most be n - (k - c.size()) + 1
  22. for i := start; i < len(nums)-(k-len(c))+1; i++ {
  23. fmt.Printf(""i = %v start = %v c = %v
  24. "", i, start, c)
  25. if i > start && nums[i] == nums[i-1] { // 这里是去重的关键逻辑,本次不取重复数字,下次循环可能会取重复数字
  26. continue
  27. }
  28. c = append(c, nums[i])
  29. generateSubsetsWithDup(nums, k, i+1, c, res)
  30. c = c[:len(c)-1]
  31. }
  32. return
  33. }