LeetCode-Python-339. 嵌套列表权重和

给定一个嵌套的整数列表，请返回该列表按深度加权后所有整数的总和。

每个元素要么是整数，要么是列表。同时，列表中元素同样也可以是整数或者是另一个列表。

示例 1:

输入: [[1,1],2,[1,1]]
输出: 10 
解释: 因为列表中有四个深度为 2 的 1 ，和一个深度为 1 的 2。

示例 2:

输入: [1,[4,[6]]]
输出: 27 
解释: 一个深度为 1 的 1，一个深度为 2 的 4，一个深度为 3 的 6。所以，1 + 4*2 + 6*3 = 27。

思路：

大问题可以拆成很多个类型相同的小问题，

所以观察到可以用递归解，所以用weight这个变量记录当前的深度（也就是权重），

当类型是数字的时候，直接返回weight和数字的乘积，

当类型是nestedList的时候，就需要对其中每一个元素进行递归处理。

class Solution(object):
    def depthSum(self, nestedList):
        """
        :type nestedList: List[NestedInteger]
        :rtype: int
        """
        def Sum(weight, l):
            if l.isInteger():
                return weight * l.getInteger()
            return sum(Sum(weight + 1, item) for item in l.getList())
        
        return sum(Sum(1, i) for i in nestedList)