您的位置:首页 > 其它

求一个集合的所有子集

2013-11-27 11:32 363 查看
最先是在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))
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: