我正在Interviewbit上解决这个问题:
给定一组不同的整数 S,返回所有可能的子集。子集中的元素必须按非降序排列。解决方案集不得包含重复的子集。
这是我简单,不言自明的代码:
ans_list=[]
def solve(i,a,tmp,l,k):
global ans_list
if i==l :
print "List to append: " ,tmp
ans_list.append(tmp)
print "ans_list currently: ", ans_list
return
j=k
while j < len(a):
tmp.append(a[j])
solve(i+1,a,tmp,l,j+1)
tmp.pop()
j+=1
class Solution:
# @param a : list of integers
# @return a list of list of integers
def subsets(self, a):
a=sorted(a)
global ans_list
tmp=[]
i=0
for l in range(len(a)+1):
solve(0,a,tmp,l,0)
return ans_list
def main():
a=[1,2,3]
ob=Solution()
li=ob.subsets(a)
print "Ans:" ,li
if __name__ == '__main__':
main()
这是输出:
List to append: []
Current ans_list: [[]]
List to append: [1]
Current ans_list: [[1], [1]]
List to append: [2]
Current ans_list: [[2], [2], [2]]
List to append: [3]
Current ans_list: [[3], [3], [3], [3]]
List to append: [1, 2]
Current ans_list: [[1, 2], [1, 2], [1, 2], [1, 2], [1, 2]]
List to append: [1, 3]
Current ans_list: [[1, 3], [1, 3], [1, 3], [1, 3], [1, 3], [1, 3]]
List to append: [2, 3]
Current ans_list: [[2, 3], [2, 3], [2, 3], [2, 3], [2, 3], [2, 3], [2, 3]]
List to append: [1, 2, 3]
Current ans_list: [[1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3]]
Ans: [[], [], [], [], [], [], [], []]
我什至尝试将ans_list
作为参数而不是全局变量传递,但结果是相同的。甚至尝试使用字典而不是列表列表,但它会导致相同的行为。但是,我注意到的一件有趣的事情是,当我在函数中将ans_list.append(tmp)
更改为ans_list.extend(tmp)
时solve
输出变为:
List to extend: []
Current ans_list: []
List to extend: [1]
Current ans_list: [1]
List to extend: [2]
Current ans_list: [1, 2]
List to extend: [3]
Current ans_list: [1, 2, 3]
List to extend: [1, 2]
Current ans_list: [1, 2, 3, 1, 2]
List to extend: [1, 3]
Current ans_list: [1, 2, 3, 1, 2, 1, 3]
List to extend: [2, 3]
Current ans_list: [1, 2, 3, 1, 2, 1, 3, 2, 3]
List to extend: [1, 2, 3]
Current ans_list: [1, 2, 3, 1, 2, 1, 3, 2, 3, 1, 2, 3]
Ans: [1, 2, 3, 1, 2, 1, 3, 2, 3, 1, 2, 3]
这几乎是意料之中的。现在我不确定在 python 中附加列表和字典时这种意外行为的原因是什么,而 extend 给出了正确的响应。任何帮助将不胜感激。
我是python的新手,使用Python 2.7.10
问题是你在 subsets() 的循环之外创建了 tmp。现在,solve() 中 tmp 的每次使用都将指向相同的列表,并且将对 ans_list 中的每个元素执行每个操作,包括 pop()。
有很多方法可以解决这个问题,但是如果您想要最少的代码更改,只需导入复制模块并更改:
ans_list.append(tmp)
自:
ans_list.append(copy.deepcopy(tmp))
编辑(我的解决方案):
ans = set()
for i in range(len(a)+1):
for j in range(len(a)+1):
ans.add(tuple(a[i:j]))