LeetCode 339. Nested List Weight Sum （嵌套列表重和）$

Given a nested list of integers, return the sum of all integers in the list weighted by their depth.

Each element is either an integer, or a list -- whose elements may also be integers or other lists.

Example 1:
Given the list [[1,1],2,[1,1]], return 10. (four 1's at depth 2, one 2 at depth 1)

Example 2:
Given the list [1,[4,[6]]], return 27. (one 1 at depth 1, one 4 at depth 2, and one 6 at depth 3; 1 + 4*2 + 6*3 = 27)

 

---

题目标签：Depth First Search

　　这道题目给了我们一个嵌套的list，在这个list里，每一个element可以是一个integer，又可以是一个list，这个list里还可以继续有list，可以无限套。所以要用到depth first search。如果能走到最里面一层呢，要利用recursive function，一层一层递归下去直到它是一个integer了，就可以返回了。所以需要另外一个function getSum。首先iterate nestedList， 把每一个element 加起来。 为了得到这个element的值，我们要把它代入getSum function， 如果这个element 是integer 直接return。 如果这个element 是一个list，那么利用相同的方法，设一个sum 把它每一个element 加起来，为了得到每一个element，把每一个element代入getSum，记得这里要把depth + 1。因为我们代入了下一层depth。

 

Java Solution:

Runtime beats 6.27% 

完成日期：07/09/2017

关键词：Depth First Search

关键点：利用递归function来实现depth first search

 

/**
 * // This is the interface that allows for creating nested lists.
 * // You should not implement it, or speculate about its implementation
 * public interface NestedInteger {
 *
 *     // @return true if this NestedInteger holds a single integer, rather than a nested list.
 *     public boolean isInteger();
 *
 *     // @return the single integer that this NestedInteger holds, if it holds a single integer
 *     // Return null if this NestedInteger holds a nested list
 *     public Integer getInteger();
 *
 *     // @return the nested list that this NestedInteger holds, if it holds a nested list
 *     // Return null if this NestedInteger holds a single integer
 *     public List<NestedInteger> getList();
 * }
 */
public class Solution 
{
    public int depthSum(List<NestedInteger> nestedList) 
    {
        int res = 0;
        
        // iterate list
        for(int i=0; i<nestedList.size(); i++)
            res += getSum(nestedList.get(i), 1);
        
        return res;
    }
    
    public int getSum(NestedInteger ele, int depth)
    {
        int sum = 0;
        
        // if ele is integer, return its value * depth;
        if(ele.isInteger())
            return depth * ele.getInteger();
        
        // if ele is a list, iterate list recursively call function;
        for(int i=0; i<ele.getList().size(); i++)
            sum += getSum(ele.getList().get(i), depth + 1);
        
        
        return sum;
    }
}

参考资料：

http://www.cnblogs.com/grandyang/p/5340305.html

 

LeetCode 算法题目列表 - LeetCode Algorithms Questions List