今天继续刷leetcode,第78题,求一个数组的所有子序列。
分析:
采用回溯的方法,每到一种情况,就将该情况加入输出结果中。
问题:
1、回溯方法;
附上c++代码:
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
vector<vector<int>> res;
vector<int> temp;
res.push_back(temp);
if(nums.empty())
return res;
sub(nums,0,res,temp);
return res;
}
void sub(vector<int>& nums,int pos,vector<vector<int>>& res,vector<int>& temp)
{
if(pos==nums.size())
return;
for(int i=pos;i<nums.size();i++)
{
temp.push_back(nums[i]);
res.push_back(temp);
sub(nums,i+1,res,temp);
temp.pop_back();
}
}
};
附上python代码:
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
res=[]
if len(nums)==0:
return []
res.append([])
self.sub(nums,0,res,[])
return res
def sub(self,nums,pos,res,temp):
if pos==len(nums):
return
for i in range(pos,len(nums)):
temp.append(nums[i])
res.append(temp.copy())
self.sub(nums,i+1,res,temp)
temp.pop()