Nested List Weight Sum
给一个数据类型叫nestedInteger, 就是嵌套, 求总和. 简单的dfs.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
class Solution { public int depthSum(List<NestedInteger> nestedList) { return dfs(nestedList, 1); } private int dfs (List<NestedInteger> nestedList, int level) { int sum = 0; for(NestedInteger n : nestedList) { if(n.isInteger()) { sum += n.getInteger() * level; }else { sum += dfs(n.getList(), level + 1); } } return sum; } } |