LintCode python 小白-简单题-22平面列表

来源:互联网 发布:淘宝的美版iphone7渠道 编辑:程序博客网 时间:2024/06/11 06:11

题目:给定一个列表,该列表中的每个要素要么是个列表,要么是整数。将其变成一个只包含整数的简单列表。

注意事项:如果给定的列表中的要素本身也是一个列表,那么它也可以包含列表。
样例:
给定 [1,2,[1,2]],返回 [1,2,1,2]。
给定 [4,[3,[2,[1]]]],返回 [4,3,2,1]。
思路:递归算法,对列表进行列表的查找。。。。

相关知识:

  1. type的运用,以后在数字的判断也会更加方便。
  2. http://blog.csdn.net/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)