Day17 平衡二叉树 + 二叉树的所有路径 + 左叶子之和

110 平衡二叉树

题目链接：
110. 平衡二叉树 - 力扣（LeetCode）

给定一个二叉树，判断它是否是平衡二叉树。

方法1 自顶向下

先序遍历

class Solution {
public:
    int depth(TreeNode* root)
    {
        if(root == nullptr) return 0;
        return max(depth(root->left), depth(root->right)) + 1;
    }

    bool isBalanced(TreeNode* root) {
        if(root == nullptr) return true;
        return abs(depth(root->left) - depth(root->right)) <=1 && isBalanced(root->left) && isBalanced(root->right);
    }
};

方法2 自底向上

参考题解：
代码随想录 (programmercarl.com)

后序遍历

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 二叉树的所有路径

题目链接：257. 二叉树的所有路径 - 力扣（LeetCode）

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

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

思路：
先序遍历，碰到叶子节点，放入到ans中。

class Solution {
public:
    void dfs(TreeNode* cur, string path, vector<string>& ans)
    {
        if(cur == nullptr)
        {
            return;
        }
        if(cur->left == nullptr && cur->right == nullptr)
        {
            if(path.empty()) path += to_string(cur->val);
            else
            {
                path += "->";
                path += to_string(cur->val);
            }
            ans.push_back(path);
        }
        if(path.empty()) path += to_string(cur->val);
        else
        {
            path += "->";
            path += to_string(cur->val);
        }
        dfs(cur->left, path, ans);
        dfs(cur->right, path, ans);
        return;
    }

    vector<string> binaryTreePaths(TreeNode* root) {
        string path;
        vector<string> ans;
        dfs(root, path, ans);
        return ans;
    }
};

代码可精简：
参考：代码随想录 (programmercarl.com)

class Solution {
public:
    void traversal(TreeNode* cur, string path, vector<string>& result) {
        path += to_string(cur->val); // 中
        if (cur->left == NULL && cur->right == NULL) {
            result.push_back(path);
            return;
        }
        if (cur->left) traversal(cur->left, path + "->", result); // 左
        if (cur->right) traversal(cur->right, path + "->", result); // 右
    }

    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;
        string path;
        if (root == NULL) return result;
        traversal(root, path, result);
        return result;
    }
};

404 左叶子之和

题目链接：
404. 左叶子之和 - 力扣（LeetCode）

给定二叉树的根节点 root ，返回所有左叶子之和。

思路：
对每个节点添加一个状态state，设定根节点state=-1，左孩子state=1，右孩子state=0，当为叶子节点，且state为1时，加入到ans中。

class Solution {
public:
    void dfs(TreeNode* cur, int state, int& ans)
    {
        if(cur == nullptr) return;
        if(cur->left == nullptr && cur->right == nullptr)
        {
            if(state == 1) ans += cur->val;
            return;
        }
        dfs(cur->left, 1, ans);
        dfs(cur->right, 0, ans);
        return;
    }

    int sumOfLeftLeaves(TreeNode* root) {
        int ans = 0;
        int state = -1;
        dfs(root, state, ans);
        return ans;
    }
};