-
Notifications
You must be signed in to change notification settings - Fork 41
/
solution.ts
33 lines (31 loc) · 1.02 KB
/
solution.ts
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
class TreeNode {
val: number
left: TreeNode | null
right: TreeNode | null
constructor (val?: number, left?: TreeNode | null, right?: TreeNode | null) {
this.val = (val === undefined ? 0 : val);
this.left = (left === undefined ? null : left);
this.right = (right === undefined ? null : right);
}
}
function levelOrder (root: TreeNode | null): number[][] {
if (root === null) {
return [];
}
const result:number[][] = [];
bfs([root, ], result, false);
return result;
}
function bfs (levelNodes:TreeNode[], result:number[][], reverse:boolean):void {
const vals:number[] = [];
const nextNodes:TreeNode[] = [];
for (let i = 0; i < levelNodes.length; i++) {
const node = levelNodes[i];
vals.push(node.val);
node.left !== null && nextNodes.push(node.left);
node.right !== null && nextNodes.push(node.right);
}
reverse && vals.reverse();
result.push(vals);
nextNodes.length > 0 && bfs(nextNodes, result, !reverse);
}