lintcode 22.平面列表

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 isinstance(nestedList, int):
            return [nestedList]

        result = []
        for ele in nestedList:
            result.append(self.flatten(ele))

        return result

'''
答案是错误的,不应该用append，应该改为extend，答案就对了
'''
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 isinstance(nestedList, int):
            return [nestedList]

        result = []
        for ele in nestedList:
            result.extend(self.flatten(ele))

        return result