hello-algo/ru/codes/javascript/chapter_tree/binary_tree_bfs.js

/**
 * File: binary_tree_bfs.js
 * Created Time: 2022-12-04
 * Author: IsChristina (christinaxia77@foxmail.com)
 */

const { arrToTree } = require('../modules/TreeNode');
const { printTree } = require('../modules/PrintUtil');

/* Обход в ширину */
function levelOrder(root) {
    // Инициализировать очередь и добавить корневой узел
    const queue = [root];
    // Инициализировать список для хранения последовательности обхода
    const list = [];
    while (queue.length) {
        let node = queue.shift(); // Извлечение из очереди
        list.push(node.val); // Сохранить значение узла
        if (node.left) queue.push(node.left); // Поместить левый дочерний узел в очередь
        if (node.right) queue.push(node.right); // Поместить правый дочерний узел в очередь
    }
    return list;
}

/* Driver Code */
/* Инициализация двоичного дерева */
// Здесь используется функция, напрямую строящая двоичное дерево из массива
const root = arrToTree([1, 2, 3, 4, 5, 6, 7]);
console.log('\nИнициализация двоичного дерева\n');
printTree(root);

/* Обход в ширину */
const list = levelOrder(root);
console.log('\nПоследовательность печати узлов при обходе в ширину = ' + list);