求一个集合的所有子集

最先是在leedCode上看到的一个题目,走了一点弯路,使用多重for循环,结果问题很多,后来换了一种思路,代码简介而且高效。题目是这样的:

Subsets:
Given a set of distinct integers, S, return all possible subsets.
Note:
Elements in a subset must be in non-descending order.
The solution set must not contain duplicate subsets.
For example,
If S = [1,2,3], a solution is:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

具体代码实现如下:

class Solution {
public:
    vector<vector<int> > subsets(vector<int> &S) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        vector<vector<int> > result;
        vector<int> elem;
        int i,j,size;
        sort(S.begin(),S.end()); //sort
        result.push_back(elem); //[]
        size=result.size();
        for(i=0;i<S.size();i++)  //添加S[i]到集合result
        {
            for(j=0;j<size;j++)  //对集合中每个元素添加S[i]形成新的元素,然后加到result尾部
            {
                elem.resize(result[j].size());
                memcpy(&elem[0], &result[j][0], elem.size()*sizeof(int));
                elem.push_back(S[i]);
                result.push_back(elem);
            }
            size=result.size();
        }
            
        return result;
    }
};

大致思路:首先结果集合result为空,原始数据集合为S;我们遍历S的元素,将每个元素添加到result的每个元素中,形成新的元素,将新元素添加到result(注:result的旧元素不能要覆盖)。大致如下:考虑三个元素的[1,2,3] -- 红色部分为新添加元素

时刻1:result=[[]], S=[1,2,3],对S[0]操作,S[0]添加到result[0]形成[1],添加到result,从而result=[[],[1]];

时刻2:result=[[], [1]],S=[1,2,3],对S[1]操作,S[1]添加到result每个元素分别形成[2],[1,2],将其添加到result中:result=[[], [1],[2], [1,2]]

时刻3:result=[[], [1], [2], [1,2]],对S[2]操作,S[2]添加到result每个元素分别形成[3], [1,3], [2,3], [1,2,3],将其添加到result中:result=[[], [1], [2], [1,2], [3], [1,3], [2,3], [1,2,3]]

此时,处理完成,result即为最后的子集集合。


使用python代码更简洁,思路和上面一样:

#!/usr/bin/python
import copy
source = [1,2,3,4,5,6,7,8,9,10]
result =[[]]
def add(sum,k):
    for i in range(len(sum)):
        elem=copy.deepcopy(sum[i])
        elem.append(k)
        sum.append(elem)

if __name__ == "__main__":
    size=len(source)
    for i in range(size):
        add(result,source[i])
    print "set:%s"%(source)
    print "subset:\n%s"%("\n".join([str(result[elem]) for elem in range(len(result))]))
    print "totalsize:%s"%(len(result))

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值