给你一个整数嵌套列表 nestedList ,每一个元素要么是一个整数,要么是一个列表(这个列表中的每个元素也同样是整数或列表)。
整数的 深度 取决于它位于多少个列表内部。例如,嵌套列表 [1,[2,2],[[3],2],1] 的每个整数的值都等于它的 深度 。令 maxDepth 是任意整数的 最大深度 。
整数的 权重 为 maxDepth - (整数的深度) + 1 。
将 nestedList 列表中每个整数先乘权重再求和,返回该加权和。
示例 1:
输入:nestedList = [[1,1],2,[1,1]]
输出:8
解释:4 个 1 在深度为 1 的位置, 一个 2 在深度为 2 的位置。
11 + 11 + 22 + 11 + 1*1 = 8
示例 2:
输入:nestedList = [1,[4,[6]]]
输出:17
解释:一个 1 在深度为 3 的位置, 一个 4 在深度为 2 的位置,一个 6 在深度为 1 的位置。
13 + 42 + 6*1 = 17
提示:
1 <= nestedList.length <= 50
嵌套列表中整数的值在范围 [-100, 100]
任意整数的最大 深度 小于等于 50
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/nested-list-weight-sum-ii
方法一:DFS
C++提交内容:
/*** // This is the interface that allows for creating nested lists.* // You should not implement it, or speculate about its implementation* class NestedInteger {* public:* // Constructor initializes an empty nested list.* NestedInteger();** // Constructor initializes a single integer.* NestedInteger(int value);** // Return true if this NestedInteger holds a single integer, rather than a nested list.* bool isInteger() const;** // Return the single integer that this NestedInteger holds, if it holds a single integer* // The result is undefined if this NestedInteger holds a nested list* int getInteger() const;** // Set this NestedInteger to hold a single integer.* void setInteger(int value);** // Set this NestedInteger to hold a nested list and adds a nested integer to it.* void add(const NestedInteger &ni);** // Return the nested list that this NestedInteger holds, if it holds a nested list* // The result is undefined if this NestedInteger holds a single integer* const vector &getList() const;* };*/class Solution {
public:int depthSumInverse(vector& nestedList) {int res = 0;vector> integersListByDepth;for (auto& nl : nestedList) {dfs(nl, 0, integersListByDepth);}int N = integersListByDepth.size();for (int i = 0; i < N; i++) {for (int j = 0; j < integersListByDepth[i].size(); j++) {res += integersListByDepth[i][j] * (N - i);}}return res;}private:void dfs(NestedInteger& ni, int depth, vector>& integersListByDepth) {if (depth == integersListByDepth.size()) {integersListByDepth.push_back(vector());}if (ni.isInteger()) {integersListByDepth[depth].push_back(ni.getInteger());} else {for (auto& nl : ni.getList()) {dfs(nl, depth + 1, integersListByDepth);}}}
};