lintcode- 22.平面表

• 题目描述

22. 平面列表

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

样例

给定 [1,2,[1,2]]，返回 [1,2,1,2]。

给定 [4,[3,[2,[1]]]]，返回 [4,3,2,1]。

挑战

请用非递归方法尝试解答这道题。

• 分析

由于最近在学习递归问题，所以仍用递归解决的此问题。用Python3 自带isinstance函数判断list元素类型，如果元素为int型，则直接记录，若为list型，则将此元素进入递归循环继续判断。

• code

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
        if not isinstance(nestedList,list):#这里主要是防止系统直接用int型元素测试代码
            return [nestedList]
        res_list = []
        self.searchList(res_list,nestedList)
        return res_list

    def searchList(self,res_List,List):
        for i in List:
            if isinstance(i, int):
                res_List.append(i)
            elif isinstance(i, list):
                self.searchList(res_List,i)