剑指 Offer 32 - III. 从上到下打印二叉树 III JavaScript 实现

问题描述

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],

3

/
9 20
/
15 7

返回其层次遍历结果：

[
[3],
[20,9],
[15,7]
]

代码

var levelOrder = function(root) {
    if (!root) return [];
    let arr = [];
    arr.push(root);
    let res = [];
    let level = 0;
    let flag = true;
    while(arr.length){
        let levelNum = arr.length;
        res[level] = [];
        while(levelNum--){
            if(flag){
                let node = arr.shift();
                res[level].push(node.val);
                node.left && arr.push(node.left);
                node.right && arr.push(node.right);
            }else if(flag == false) {
                let node = arr.pop();
                res[level].push(node.val);
                node.right && arr.unshift(node.right);
                node.left && arr.unshift(node.left);
            }          
        }  
        flag = !flag; 
        level++;
    }
    return res;
};