Repository navigation
Expand file tree
/
Copy path364_Nested_List_Weight_Sum_II.swift
More file actions
56 lines (53 loc) · 2.06 KB
/
Copy path364_Nested_List_Weight_Sum_II.swift
File metadata and controls
56 lines (53 loc) · 2.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/**
* // This is the interface that allows for creating nested lists.
* // You should not implement it, or speculate about its implementation
* class NestedInteger {
* // Return true if this NestedInteger holds a single integer, rather than a nested list.
* public func isInteger() -> Bool
*
* // 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
* public func getInteger() -> Int
*
* // Set this NestedInteger to hold a single integer.
* public func setInteger(value: Int)
*
* // Set this NestedInteger to hold a nested list and adds a nested integer to it.
* public func add(elem: NestedInteger)
*
* // 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
* public func getList() -> [NestedInteger]
* }
*/
import Foundation
class Solution {
func depthSumInverse(_ nestedList: [NestedInteger]) -> Int {
var maxDepth = getMaxDepth(nestedList)
let weightedSum = getWeightedSum(nestedList, maxDepth)
return weightedSum
}
private func getMaxDepth(_ nestedList: [NestedInteger]) -> Int {
var currentMaxDepth = 0
for nestedInt in nestedList {
if nestedInt.isInteger() {
currentMaxDepth = max(currentMaxDepth, 1)
} else {
var currentDepth = getMaxDepth(nestedInt.getList())
currentMaxDepth = max(currentMaxDepth, currentDepth + 1)
}
}
return currentMaxDepth
}
private func getWeightedSum(_ nestedList: [NestedInteger], _ depth: Int) -> Int {
var currentSum = 0
for nestedInt in nestedList {
if nestedInt.isInteger() {
currentSum += nestedInt.getInteger() * depth
} else {
currentSum += getWeightedSum(nestedInt.getList(), depth - 1)
}
}
return currentSum
}
}