力扣(LeetCode)364. 加权嵌套序列和 II(2022.12.31)
创始人
2024-05-03 22:05:06
0次

给你一个整数嵌套列表 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);}}}
};

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...