339. Nested List Weight Sum

nested_list_weight_sum.py

class Solution:
    def depthSum(self, nestedList: List[NestedInteger]) -> int:
        
        def depth_sum_recursive(current_item, depth):
            if current_item.isInteger():
                return current_item.getInteger() * depth

            return sum(depth_sum_recursive(item, depth + 1) for item in current_item.getList())

        return sum(depth_sum_recursive(item, 1) for item in nestedList)