给定一个列表,该列表中的每个要素要么是个列表,要么是整数。将其变成一个只包含整数的简单列表。
注意事项
如果给定的列表中的要素本身也是一个列表,那么它也可以包含列表。
因为是列表就第一时间想到用python做,只要简单的用一个递归就可以了
class Solution(object):
def flatten(self, nestedList):
# Write your code here
if not nestedList:
return []
self.ajs = []
if type(nestedList) != list:
self.ajs.append(nestedList)
else:
self.my_flatten(nestedList)
return self.ajs
def my_flatten(self, L):
if not L:
return
for tmp in L:
if type(tmp) == list:
self.my_flatten(tmp)
else:
self.ajs.append(tmp)