Tony's Log

Algorithms, Distributed System, Machine Learning

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

An intuitive DFS one.

class Solution(object):
    def go(self, nList, dep):
        ret = 0
        for i in range(len(nList)):
            if(nList[i].isInteger()):
                ret += dep * (nList[i].getInteger())
            else:
                ret += self.go(nList[i].getList(), dep + 1)
        return ret

    def depthSum(self, nestedList):
        return self.go(nestedList, 1)
posted on 2016-03-31 00:38  Tonix  阅读(129)  评论(0编辑  收藏  举报