Leetcode算法学习日志-78 Subsets

来源:互联网 发布:私密空间软件 编辑:程序博客网 时间:2024/06/02 01:22

Leetcode 78 Subsets

题目原文

Given a set of distinct integers,nums, return all possible subsets (the power set).

Note: The solution set must not contain duplicate subsets.

For example,
If nums = [1,2,3], a solution is:

[  [3],  [1],  [2],  [1,2,3],  [1,3],  [2,3],  [1,2],  []]

题意分析

给一组数,输出其所有无序子数组,空子数组也考虑在内。

解法分析

本题很容易想到写出一棵子集树,用回溯的方法解,只是没有任何剪枝函数,仅仅深度优先遍历这个子集树,C++代码如下:

class Solution {private:    vector<int> temp;    vector<vector<int>> res;    vector<int> nums;    int n;public:    void DFS(int k){        if(k>n){            res.push_back(temp);            return;        }        DFS(++k);        k--;        temp.push_back(nums[k-1]);        DFS(++k);        temp.pop_back();    }    vector<vector<int>> subsets(vector<int>& numbers) {        nums=numbers;        n=nums.size();        DFS(1);        return res;      }};
注意由于子集树是一个二叉树,因此DFS内部不需要用for循环遍历同层,只需要先后进行两次递归调用DFS,用于选择0和1,对应不取或取相应数字。上述算法和backtracking类似。

本题还有一个比较巧妙的方法,同样是通过那棵二叉树找到灵感。用0-2^n-1这几个数作为最终输出中的2^n个vector的序号,对序号所表示数的二进制进行检测,某一位为1代表该vector拥有对应的一个数,依次将数填入res中。C++代码如下:

class Solution {public:    vector<vector<int>> subsets(vector<int>& nums) {        int n=nums.size();        int m=(int)pow(2.0,n);        vector<vector<int>> res(m,vector<int>());        int i,j;        for(i=0;i<n;i++){            for(j=0;j<m;j++){                if((j>>i)&1)                    res[j].push_back(nums[i]);            }        }        return res;            }};





原创粉丝点击