题目:给定一个列表,该列表中的每个要素要么是个列表,要么是整数。将其变成一个只包含整数的简单列表。
注意事项:如果给定的列表中的要素本身也是一个列表,那么它也可以包含列表。
样例:
给定 [1,2,[1,2]],返回 [1,2,1,2]。
给定 [4,[3,[2,[1]]]],返回 [4,3,2,1]。
思路:递归算法,对列表进行列表的查找。。。。
相关知识:
- type的运用,以后在数字的判断也会更加方便。
- http://blog.youkuaiyun.com/jcjc918/article/details/9623109
本次用到了类的知识,特别在self.b
代码:
class Solution(object):
# @param nestedList a list, each element in the list
# can be a list or integer, for example [1,2,[1,2]]
# @return {int[]} a list of integer
def flatten(self, nestedList):
# Write your code here
a=nestedList
if a is None: #判断是否为空
return []
self.b=[]
if type(a)!=list:#type()进行类型的判断
self.b.append(a)
else:
self.digui(a)
return self.b
def digui(self,c):
if c is None: #判断是否是[]情况
retur
for d in c:
if type(d) == list: #发现列表内存在列表
self.digui(d)
else: #不是列表代表是数字,将数字放于self.b的后面
self.b.append(d)