339 Nested List Weight Sum
339. Nested List Weight Sum
题目: https://leetcode.com/problems/nested-list-weight-sum/
难度: Easy
思路：
一开始没认真读题，直接上手开写：
class Solution(object):
def depthSum(self, nestedList):
"""
:type nestedList: List[NestedInteger]
:rtype: int
"""
def dfs(nestedList):
for item in nestedList:
if item.isInteger():
self.res += item.getInteger()
else:
dfs(item.getList())
self.res = 0
dfs(nestedList)
return self.res
然后注意到要weight by its depth.
AC
class Solution(object):
def depthSum(self, nestedList):
"""
:type nestedList: List[NestedInteger]
:rtype: int
"""
def dfs(nestedList,depth):
for item in nestedList:
if item.isInteger():
self.res += item.getInteger() * depth
else:
dfs(item.getList(), depth+1)
self.res = 0
dfs(nestedList,1)
return self.res
