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.

答案

class Solution {
    public int depthSum(List<NestedInteger> nestedList) {
        return recur(nestedList, 1);
    }
    
    public int recur(List<NestedInteger> nestedList, int depth) {
        int total = 0;
        for(int i = 0; i < nestedList.size(); i++) {
            NestedInteger ni = nestedList.get(i);
            if(ni.isInteger()) {
                total += depth * ni.getInteger();
            }
            else {
                total += recur(ni.getList(), depth + 1);
            }
        }
        return total;
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容