-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathP_339.java
More file actions
38 lines (29 loc) · 871 Bytes
/
Copy pathP_339.java
File metadata and controls
38 lines (29 loc) · 871 Bytes
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
package leetcode.easy;
import java.util.List;
public class P_339 {
private interface NestedInteger {
boolean isInteger();
Integer getInteger();
void setInteger(int value);
void add(NestedInteger ni);
List<NestedInteger> getList();
}
public int depthSum(List<NestedInteger> nestedList) {
int temp = 0;
for (int i = 0; i < nestedList.size(); i++) {
temp += dfs(nestedList.get(i), 1);
}
return temp;
}
private static int dfs(NestedInteger curr, int depth) {
if (curr.isInteger()) {
return curr.getInteger() * depth;
}
int temp = 0;
final List<NestedInteger> list = curr.getList();
for (int i = 0; i < list.size(); i++) {
temp += dfs(list.get(i), depth + 1);
}
return temp;
}
}