LintCode:平面列表

LintCode:平面列表

递归方式

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 nestedList:
            return []
        self.ans = []
        if type(nestedList) != list:
            self.ans.append(nestedList)
        else:
            self.my_flatten(nestedList)
        return self.ans

    def my_flatten(self, L):
        if not L:
            return
        for tmp in L:
            if type(tmp) == list:
                self.my_flatten(tmp)
            else:
                self.ans.append(tmp)