-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsolution.js
41 lines (32 loc) · 838 Bytes
/
solution.js
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
class TreeNode {
constructor(val, left, right) {
this.val = (val === undefined ? 0 : val);
this.left = (left === undefined ? null : left);
this.right = (right === undefined ? null : right);
}
}
/**
* @param {TreeNode} root
* @return {number[][]}
*/
const levelOrder = (root) => {
if (null === root) return [];
let list = [[root.val]], queue = [root];
while (queue.length > 0) {
let l = queue.length, singleLevel = [];
while (l > 0) {
let curr = queue.shift(), left = curr.left, right = curr.right;
if (null !== left) {
queue.push(left);
singleLevel.push(left.val);
}
if (null !== right) {
queue.push(right);
singleLevel.push(right.val);
}
l--;
}
if (singleLevel.length > 0) list.push(singleLevel);
}
return list;
};