1. 程式人生 > >Leetcode 78. 子集 dfs

Leetcode 78. 子集 dfs

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

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

示例:

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

求子集問題可以轉化成從集合中依次獲取0-n個元素的過程。。。

然後呼叫n次dfs來向集合中獲取元素。。。

程式碼如下:

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
            vector<vector<int>>ve;
            vector<int> v1;
            int Size=nums.size();
            ve.push_back(v1);
            if(Size==0)
                return ve;
            for (int i=1;i<=Size;i++)
                dfs (0,i,v1,0,ve,nums,Size);
            return ve;
    }
    void dfs (int num,int target,vector<int>& v1,int loc,vector<vector<int>>& ve,vector<int>& v,int Size)
    {
         if(num==target)
         {
             ve.push_back(v1);
             return;
         }
         for (int i=loc;i<Size;i++)
         {
                 v1.push_back(v[i]);
                 dfs (num+1,target,v1,i+1,ve,v,Size);
                 v1.pop_back();
         }
    }
    
};