Skip to content

Commit 0270892

Browse files
refactor 339
1 parent 55c0398 commit 0270892

File tree

1 file changed

+26
-27
lines changed
  • src/main/java/com/fishercoder/solutions

1 file changed

+26
-27
lines changed

src/main/java/com/fishercoder/solutions/_339.java

Lines changed: 26 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -5,37 +5,36 @@
55
import java.util.List;
66

77
public class _339 {
8-
class SolutionWithGlobalSum {
9-
private int sum = 0;
8+
public static class Solution1 {
9+
private int sum = 0;
1010

11-
public int depthSum(List<NestedInteger> nestedList) {
12-
return dfs(nestedList, 1);
13-
}
14-
15-
private int dfs(List<NestedInteger> nestedList, int depth) {
16-
for (NestedInteger ni : nestedList) {
17-
if (ni.isInteger()) {
18-
sum += depth * ni.getInteger();
19-
} else {
20-
dfs(ni.getList(), depth + 1);
21-
}
22-
}
23-
return sum;
24-
}
11+
public int depthSum(List<NestedInteger> nestedList) {
12+
return dfs(nestedList, 1);
2513
}
2614

27-
class SolutionWithLocalSum {
28-
public int depthSum(List<NestedInteger> nestedList) {
29-
return dfs(nestedList, 1);
15+
private int dfs(List<NestedInteger> nestedList, int depth) {
16+
for (NestedInteger ni : nestedList) {
17+
if (ni.isInteger()) {
18+
sum += depth * ni.getInteger();
19+
} else {
20+
dfs(ni.getList(), depth + 1);
3021
}
22+
}
23+
return sum;
24+
}
25+
}
3126

32-
private int dfs(List<NestedInteger> nestedList, int depth) {
33-
int sum = 0;
34-
for (NestedInteger ni : nestedList) {
35-
sum += ni.isInteger() ? depth * ni.getInteger() : dfs(ni.getList(), depth + 1);
36-
}
37-
return sum;
38-
}
27+
public static class Solution2 {
28+
public int depthSum(List<NestedInteger> nestedList) {
29+
return dfs(nestedList, 1);
3930
}
4031

41-
}
32+
private int dfs(List<NestedInteger> nestedList, int depth) {
33+
int sum = 0;
34+
for (NestedInteger ni : nestedList) {
35+
sum += ni.isInteger() ? depth * ni.getInteger() : dfs(ni.getList(), depth + 1);
36+
}
37+
return sum;
38+
}
39+
}
40+
}

0 commit comments

Comments
 (0)