-
Notifications
You must be signed in to change notification settings - Fork 41
/
solution.ts
32 lines (30 loc) · 1022 Bytes
/
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
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);
return result;
}
function bfs (levelNodes:TreeNode[], result:number[][]):void {
const levelVals:number[] = [];
const nextLevelNodes:TreeNode[] = [];
for (let i = 0; i < levelNodes.length; i++) {
const node:TreeNode = levelNodes[i];
levelVals.push(node.val);
node.left !== null && nextLevelNodes.push(node.left);
node.right !== null && nextLevelNodes.push(node.right);
}
result.push(levelVals);
nextLevelNodes.length && bfs(nextLevelNodes, result);
}