DAY17：平衡二叉树110、二叉树所有路径257、左叶子之和404

Leetcode: 110 平衡二叉树

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

递归法

先判断左子树高度是不是为-1，再判断右边。然后计算左右子树相减是不是大于1，如果大于了就返回-1.如果没有多余，就是当前节点+左右子树的最大高度。

class Solution {
public:
    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;

        int result;
        if (abs(leftheight - rightheight) > 1) {  // 中
            result = -1;
        } else {
            result = 1 + max(leftheight, rightheight); 
        }
        return result;
        
    }
    bool isBalanced(TreeNode* root) {
        return getheight(root) == -1 ? false : true;
    }
};

Leetcode: 257 二叉树的所有路径

本题要求寻找二叉树的所有路径，因此需要深度搜索。因此采用前序遍历的方法。但同时本题还需要一个回溯的过程，因为我们需要记录所有遍历过的路径。参考代码随想录如下：

代码随想录

递归法

• 确定递归参数和返回值：要传入根节点，记录每一条路径的path，和存放结果集的result，这里递归不需要返回值。
• 确定递归终止条件：遇到叶子节点，就需要开始收集路径，保存结果。
• 确定单层递归逻辑：按照中左右的顺序，判断左右节点是否存在，如果存在就递归，然后回溯。

这里需要注意的一点，虽然没有写出回溯的代码，但是回溯过程包含在代码中，因为定义string path的时候没有加上&，因此对path的每次操作都是赋值操作，不改变传入的path，因此实现了回溯的过程。

class Solution {
public:
    void getroutine(TreeNode* node, string path, vector<string>& result){//注意定义函数的时候要不要带上路径
        path += to_string(node->val);//将值转换为字符串格式
        if(node->left == NULL && node->right == NULL){//判断是不是叶子节点。
            result.push_back(path);//保存路径
            return;
        } 
        if(node->left) getroutine(node->left, path + "->", result);//递归
        if(node->right) getroutine(node->right, path + "->", result);
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        string path;
        vector<string> result;
        if(root == NULL) return result;//先要判断一下根节点为NULL的情况。
        getroutine(root, path, result);
        return result;//返回结果
    }
};

迭代法

需要两个栈来实现，一个栈用于前序遍历，另一个栈用路径的变化

class Solution {
public:
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;
        stack<string> pathst;//定义保存路径的栈
        stack<TreeNode*> st;//定义二叉树本身的栈
        if(root == NULL) return result;//先判断特殊情况
        st.push(root);//根节点入栈
        pathst.push(to_string(root->val));
        while(!st.empty()){
            TreeNode* node = st.top();
            st.pop();
            string path = pathst.top();
            pathst.pop();
            if(node->left == NULL && node->right == NULL){
                result.push_back(path);//碰到叶子节点，就需要收集路径
            }
            if(node->left){
                st.push(node->left);
                pathst.push(path + "->" + to_string(node->left->val));//将路径入栈
            }
            if(node->right){
                st.push(node->right);
                pathst.push(path + "->" + to_string(node->right->val));
            }
        }
        return result;
    }
};

Leetcode: 404 左叶子之和

判断左叶子的条件

该节点的左孩子不为空，左孩子的左右孩子为空，则左孩子为左叶子。

按照左右中的顺序为后序遍历

递归法

• 输入节点，输出左叶子之和
• 终止条件如上
• 递归逻辑，前序遍历。

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if(root == NULL) return 0;
        if(root->left == NULL && root->right== NULL) return 0;
        
        int leftnum, rightnum;
        leftnum = sumOfLeftLeaves(root->left);
        if(root->left != NULL && root->left->left == NULL && root->left->right == NULL){
            leftnum = root->left->val;
        }
        rightnum = sumOfLeftLeaves(root->right);
        int sum = leftnum + rightnum;
        return sum;

    }
};

迭代法

前中后序都可以，判断条件同上。

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        stack<TreeNode*> st;
        if (root == NULL) return 0;
        st.push(root);
        int result = 0;
        while (!st.empty()) {
            TreeNode* node = st.top();
            st.pop();
            if (node->left != NULL && node->left->left == NULL && node->left->right == NULL) {
                result += node->left->val;
            }
            if (node->right) st.push(node->right);
            if (node->left) st.push(node->left);
        }
        return result;
    }
};