Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content

Commit 9ed098f

Browse files
refactor 637
1 parent 30f15e2 commit 9ed098f

File tree

1 file changed

+24
-22
lines changed
  • src/main/java/com/fishercoder/solutions

1 file changed

+24
-22
lines changed

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

Lines changed: 24 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -29,31 +29,33 @@
2929
*/
3030
public class _637 {
3131

32-
public List<Double> averageOfLevels(TreeNode root) {
33-
List<Double> result = new ArrayList<>();
34-
if (root == null) {
35-
return result;
36-
}
37-
Queue<TreeNode> queue = new LinkedList<>();
38-
queue.offer(root);
39-
while (!queue.isEmpty()) {
40-
int size = queue.size();
41-
double sum = 0.0;
42-
for (int i = 0; i < size; i++) {
43-
TreeNode curr = queue.poll();
44-
if (curr != null) {
45-
sum += curr.val;
46-
}
47-
if (curr.left != null) {
48-
queue.offer(curr.left);
49-
}
50-
if (curr.right != null) {
51-
queue.offer(curr.right);
32+
public static class Solution1 {
33+
public List<Double> averageOfLevels(TreeNode root) {
34+
List<Double> result = new ArrayList<>();
35+
if (root == null) {
36+
return result;
37+
}
38+
Queue<TreeNode> queue = new LinkedList<>();
39+
queue.offer(root);
40+
while (!queue.isEmpty()) {
41+
int size = queue.size();
42+
double sum = 0.0;
43+
for (int i = 0; i < size; i++) {
44+
TreeNode curr = queue.poll();
45+
if (curr != null) {
46+
sum += curr.val;
47+
}
48+
if (curr.left != null) {
49+
queue.offer(curr.left);
50+
}
51+
if (curr.right != null) {
52+
queue.offer(curr.right);
53+
}
5254
}
55+
result.add(sum / size);
5356
}
54-
result.add(sum / size);
57+
return result;
5558
}
56-
return result;
5759
}
5860

5961
}

0 commit comments

Comments
 (0)