Advertisement
smj007

Nested List Weight Sum

Aug 6th, 2024
225
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.42 KB | None | 0 0
  1. class Solution:
  2.     def depthSum(self, nestedList: List[NestedInteger]) -> int:
  3.         return self.helper(nestedList)
  4.    
  5.     def helper(self, nestedList, depth=1):
  6.         depthSum = 0
  7.         for element in nestedList:
  8.             if element.isInteger():
  9.                 depthSum += depth*element.getInteger()
  10.             else:
  11.                 depthSum += self.helper(element.getList(), depth+1)
  12.         return depthSum
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement