Binary Tree Array - children

getChildrenByNode.js

// Binary tree represented as an array
//       1
//      / \
//     2   3
//    / \   \
//   4   5   6

const tree = [1, 2, 3, 4, 5, null, 6];


function getChildrenByNode(tree, node) {
    for (let i = 0; i < tree.length; i++) {
        if(tree[i] === node) {
            let results = [];
            
            if(2 * i < tree.length) {
                results.push(tree[2*i + 1]);
            }
            if((2 * i + 1) < tree.length) {
                results.push(tree[2*i + 2]);
            }

            return results;
        }
    }
}


console.log(getChildrenByNode(tree, 2));
console.log(getChildrenByNode(tree, 3));