【前端er每日算法】110平衡二叉树、257二叉树所有路径、404左叶子之和

89 阅读2分钟

题目一 110. 平衡二叉树

给定一个二叉树,判断它是否是高度平衡的二叉树。

本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

思路

左子树与右子树高度差小于等于1,并且子树上的每个节点都符合这个条件,求左右子树的高度,如果左右子树高度差值已经大于1了,则返回-1,标记当前节点已经不是平衡二叉树了,不用返回真实的高度,如果符合条件的话则返回正常的高度,最后判断根节点高度是否等于-1。


var getHeight = function(root) {
    if (!root) {
        return 0;
    }
    const leftHeight = getHeight(root.left);
    if (leftHeight === -1) {
        return -1;
    }
    const rightHeight = getHeight(root.right);
    if (rightHeight === -1) {
        return -1;
    }
    return Math.abs(leftHeight - rightHeight) > 1 ? -1 : Math.max(leftHeight, rightHeight) + 1;
}
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
var isBalanced = function(root) {
    if (!root) {
        return true;
    }
    const height = getHeight(root);
    return height === -1 ? false : true;
};

var isBalanced = function(root) {
    const getHeight = root => {
        if (!root) {
            return 0;
        }
        const leftHeight = getHeight(root.left);
        const rightHeight = getHeight(root.right);
        if (leftHeight === -1 || rightHeight === -1
            || Math.abs(leftHeight - rightHeight) > 1) {
            return -1;
        }
        return Math.max(leftHeight, rightHeight) + 1;
    }
    return getHeight(root) !== -1
};

题目二 257. 二叉树的所有路径

给你一个二叉树的根节点 root ,按 任意顺序 ,返回所有从根节点到叶子节点的路径。

叶子节点 是指没有子节点的节点。

思路

学了回溯法后,发现这个就跟回溯很像了,遍历到当前节点,放进path,然后访问左、右节点,如果节点处理完了,要回溯最后一个节点。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {string[]}
 */
var binaryTreePaths = function(root) {
    const result = [];
    const path = [];
    const getPath = root => {
        path.push(root.val);
        if (!root.left && !root.right) {
            result.push(path.join('->'));
            return;
        }
        if (root.left) {
            getPath(root.left);
            path.pop();
        }
        if (root.right) {
            getPath(root.right);
            path.pop();
        }
    }
    getPath(root);
    return result;
};

题目三 404. 左叶子之和

思路

解法1. 记录是否是左边的叶子节点标记,如果遇到叶子节点,并且是路径为左的节点,则加入。

解法2. 题解的解法是如果是叶子节点并且是当前父节点的左孩子,则将其加入总和,所以为了判断是否是做孩子,要把父节点传入函数。

/**
 * @param {TreeNode} root
 * @return {number}
 */
var sumOfLeftLeaves = function(root) {
    let sum = 0;
    const getSum = (root, isLeft) => {
        if (!root) {
            return;
        }
        if (!root.left && !root.right && isLeft) {
            sum += root.val;
        }
        if (root.left) {
            getSum(root.left, true);
        }
        if (root.right) {
            getSum(root.right, false);
        }
    }
    getSum(root, false)
    return sum;
};

// 左孩子判断法
var sumOfLeftLeaves = function(root) {
    let sum = 0;
    const traversal = (root, pre) => {
        if (!root) {
            return;
        }
        if (!root.left && !root.right && root === pre.left) {
            sum += root.val;
        }
        traversal(root.left, root);
        traversal(root.right, root);
    };
    traversal(root, root);
    return sum;
}