339. Nested List Weight Sum

[Shawngbk]

/**

• // 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 getList();

• } */ public class Solution { public int depthSum(List nestedList) { return helper(nestedList, 1); }

  private int helper(List nestedList, int level) { int sum = 0; for(NestedInteger temp : nestedList) { if(temp != null) { if(temp.isInteger()) { sum = sum + temp.getInteger()*level; } else { sum = sum + helper(temp.getList(), level+1); } } } return sum; } }

[Shawngbk]

https://discuss.leetcode.com/topic/60924/java-solution-dfs-and-bfs

[Shawngbk]

Linkedin