代码随想录算法训练营第十七天 | 110.平衡二叉树 257. 二叉树的所有路径 404.左叶子之和

110.平衡二叉树

求二叉树的深度的使用前序遍历，求二叉树的高度，使用后序遍历，后序遍历就是自底向上，一步一步往上，高度加一

class Solution {
public:
    bool flag = true;
    bool isBalanced(TreeNode* root) 
    {
        int h = getheight(root);
        return flag;
    }

    int getheight(TreeNode* node)
    {
        if (node == nullptr) return 0;

        int leftheight = getheight(node->left);
        int rightheight = getheight(node->right);

        flag = flag && abs(leftheight - rightheight) <= 1;

        return max(leftheight, rightheight) + 1;

    }
};

使用后序遍历，递归求每个节点的高度，同时额外维护一个flag

官方逻辑中，当左右子树的高度差>1时，直接返回-1

class Solution {
public:
    // 返回以该节点为根节点的二叉树的高度，如果不是平衡二叉树了则返回-1
    int getHeight(TreeNode* node) {
        if (node == NULL) {
            return 0;
        }
        int leftHeight = getHeight(node->left);
        if (leftHeight == -1) return -1;
        int rightHeight = getHeight(node->right);
        if (rightHeight == -1) return -1;
        return abs(leftHeight - rightHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
    }
    bool isBalanced(TreeNode* root) {
        return getHeight(root) == -1 ? false : true;
    }
};

257. 二叉树的所有路径

初见想法：无

class Solution {
public:
    vector<string> ans;
    vector<string> binaryTreePaths(TreeNode* root) {
        if (root == nullptr) return ans;
        traversal(root, to_string(root->val));

        return ans;
    }

    void traversal(TreeNode* node, string s)
    {
        if (node->left == nullptr && node->right == nullptr)
        {
            ans.push_back(s);
            return;
        }

        if (node->left) traversal(node->left, s + "->" + to_string(node->left->val));
        if (node->right) traversal(node->right, s + "->" + to_string(node->right->val));

    }
};

重点在于理清traversal两个变量之间的关系，当遍历至这一个node的时候就需要在s里加上其值，而不必在function里面操作，traversal里面主要是判断是否需要push以及往下遍历。

404.左叶子之和

初见想法：二叉树的层序遍历，不断把左边的值加上

class Solution {
public:
    int leftsum = 0;
    int sumOfLeftLeaves(TreeNode* root) {

        traversal(root);
        return leftsum;

    }

    void traversal(TreeNode* node)
    {
        if (node == nullptr) return;

        if (node->left && node->left->left == nullptr && node->left->right == nullptr) leftsum += node->left->val;

        traversal(node->left);
        traversal(node->right);
    }    


};

不是层序遍历，简单的遍历，但是需要注意的是左叶子节点的定义为上一个节点的左节点同时此节点的左右子节点都为空节点。