LeetCode 364. 加权嵌套序列和 II（重复叠加）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给一个嵌套整数序列，请你返回每个数字在序列中的加权和，它们的权重由它们的深度决定。

序列中的每一个元素要么是一个整数，要么是一个序列（这个序列中的每个元素也同样是整数或序列）。

与前一个问题不同的是，前一题的权重按照从根到叶逐一增加，而本题的权重从叶到根逐一增加。

也就是说，在本题中，叶子的权重为1，而根拥有最大的权重。

示例 1:
输入: [[1,1],2,[1,1]]
输出: 8 
解释: 四个 1 在深度为 1 的位置， 一个 2 在深度为 2 的位置。示例 2:
输入: [1,[4,[6]]]
输出: 17 
解释: 一个 1 在深度为 3 的位置， 一个 4 在深度为 2 的位置，
一个 6 在深度为 1 的位置。 1*3 + 4*2 + 6*1 = 17。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/nested-list-weight-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：
LeetCode 339. 嵌套列表权重和（DFS）
LeetCode 5363. 做菜顺序 hard

/*** // 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<NestedInteger> &getList() const;* };*/
class Solution {
public:int depthSumInverse(vector<NestedInteger>& nestedList) {int presum = 0, ans = 0, i;vector<NestedInteger> nextLevel;for(i = 0; i < nestedList.size(); ++i){if(nestedList[i].isInteger())presum += nestedList[i].getInteger();else{auto temp = nestedList[i].getList();for(auto& t : temp)nextLevel.push_back(t);}if(i == nestedList.size()-1){ans += presum;swap(nestedList,nextLevel);nextLevel.clear();i = -1;}}return ans;}
};