Function levelOrder0

binary_tree_level_order_traversal_102/solution.go:45–81

Note: study again.

(root *TreeNode)

// Note: study again.
func levelOrder0(root *TreeNode) [][]int {
	levels := make([][]int, 0)
	if root == nil {
		return levels
	}

	levelCnt := 1
	q := []*TreeNode{root}
	for len(q) > 0 {
		// dequeue and create level levelCnt times
		newLevelCnt := 0
		level := make([]int, 0)
		for i := 0; i < levelCnt; i++ {
			// add nodes value to the level
			n := q[0]
			level = append(level, n.Val)

			if n.Left != nil {
				q = append(q, n.Left)
				newLevelCnt++
			}

			if n.Right != nil {
				q = append(q, n.Right)
				newLevelCnt++
			}

			// dequeue the node
			q = q[1:]
		}

		levels = append(levels, level)
		levelCnt = newLevelCnt
	}

	return levels
}