LeetCode - 子集II

題目鏈接:https://leetcode-cn.com/problems/subsets-ii/

題目描述

給定一組可能包含重複元素的整數數組 nums,返回該數組所有可能的子集(冪集)。

說明: 解集不能包含重複的子集。

示例:

輸入: nums = [1,2,2]
輸出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

我的思路

  • 是LeetCode-子集的加強版,增加了一些判斷條件,整體的思路沒變。
  • 同樣使用回溯法,屬於沒有遞歸基的遞歸,每調用一次遞歸,就輸出當前 temp 中的元素。例如本題,元素彈出的順序爲 1 | 1,2 | 1,2,2 | 1,2 | 2,2 | 2 。遞歸樹如下:
    root
  /	  |	  \
 1    2    2
/ \   |
2 2   2
|
3
  • 但我們會發現,1,2 數組有兩個,不符題意。因此每遞歸到一層,應該去掉該層相同的數字,比如例子中該去掉第二層的第二個2。
  • 還有一點不同,需要在運算前,將原數組進行排序。

代碼如下:

class Solution {
    vector<vector<int>> res;
    vector<int> temp;
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        recur(nums, 0);
        return res;
    }
    void recur(vector<int>& nums, int index){
        res.push_back(temp);
        for(int i = index; i < nums.size(); i++){
            if(i > index && nums[i] == nums[i-1]) continue;	//去掉該層相同的數字
            temp.push_back(nums[i]);
            recur(nums, i + 1);
            temp.pop_back();
        }
        return;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章