代码随想录算法训练营第十五天 ｜ 102.二叉树的层序遍历、226.翻转二叉树、101. 对称二叉树

LeetCode 102.二叉树的层序遍历

题目链接🔗：102.二叉树的层序遍历

解题思路🤔

用队列记录二叉树中的元素，让其按层依次进入、离开队列，离开队列后存入数组即可。

遇到的问题😢

无

代码实现👨🏻‍💻

第一种方法

var levelOrder = function(root) {
    const res = [], queue = [];
    queue.push(root);
    
    if(!root) return res;
    
    while(queue.length !== 0) {
        let len = queue.length; // 记录当前层级节点数
        let curLevel = []; // 存放每一层的节点 
        
        for(let i = 0; i < len; i++) {
            let node = queue.shift();
            curLevel.push(node.val);
            
            // 存放当前层下一层的节点
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.push(curLevel); //把每一层的结果放到结果数组
    }
    return res;
};

总结📖

---

LeetCode 226.翻转二叉树

题目链接🔗：226.翻转二叉树

解题思路🤔

整体思路是反转每个节点的子节点。注意想清楚使用的是哪种遍历。

遇到的问题😢

无

代码实现👨🏻‍💻

递归版本前序遍历

var invertTree = function(root) {
    if (!root) return null;
    // 交换左右节点
    const rightNode = root.right;
    root.right = invertTree(root.left);
    root.left = invertTree(rightNode);
    return root;
};

层序遍历

var invertTree = function(root) {
    // 节点交换函数
    const invertNode = function(root, left, right) {
        let temp = left;
        left = right;
        right = temp;
        root.left = left;
        root.right = right;
    }
    // 使用层序遍历
    let queue = [];
    
    if(!root) return root;
    queue.push(root);
    
    while(queue.length) {
        let length = queue.length;
        while(length--) {
            let node = queue.shift();
            // 交换节点
            invertNode(node, node.left, node.right);
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
    }
    return root;
};

总结📖

---

LeetCode 101. 对称二叉树

题目链接🔗：101. 对称二叉树

解题思路🤔

基本思路是判断根节点的左右子树是否是互相翻转的。进一步理解，比较的是两个子树里侧与外侧的元素是否相等。

在递归遍历中，左右两个子树也需要遍历，左子树的遍历顺序是左右中，右子树的遍历顺序是右左中。

遇到的问题😢

代码实现👨🏻‍💻

第一种方法

var isSymmetric = function(root) {
    // 1. 确定递归的参数 root.left root.right和返回值true false 
    const compareNode = function(left, right) {
        // 2. 确定终止条件 空的情况
        if(left === null && right !== null || left !== null && right === null) {
            return false;
        } else if(left === null && right === null) {
            return true;
        } else if(left.val !== right.val) {
            return false;
        }
        // 3. 确定单层递归逻辑
        let outSide = compareNode(left.left, right.right);
        let inSide = compareNode(left.right, right.left);
        return outSide && inSide;
    }
    if(root === null) {
        return true;
    }
    return compareNode(root.left, root.right);
};

总结📖

今日收获