【小白刷题之路Day26】令人虎躯一震的代码

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

说明:解集不能包含重复的子集。

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = [[]]
        for num in nums:
            res = res + [i + [num] for i in res]
        return res

作者:powcai
链接:https://leetcode-cn.com/problems/subsets/solution/hui-su-suan-fa-by-powcai-5/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处

思考良久仍不得要领,答主借助python语法特性短短四行解决,不过短短四行真是值得细细揣摩,太厉害了!

(对应的递归法见这篇文章

对应的位运算法更是跪了,用二进制01来代表是否选择该位置元素。

nums = [1,2,3]

000, 001, 010, 011, 100, 101, 110, 111, 0~7这8个数字二进制完美对应了子集的元素选择。这个思路给跪了!!!  膜拜~~~

我的提交:

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

 

posted @ 2019-09-25 15:28  ACStrive  阅读(233)  评论(0编辑  收藏  举报