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

110.平衡二叉树

题目
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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;
    }
};

迭代法

• 在104.二叉树的最大深度 (opens new window)中我们可以使用层序遍历来求深度，但是就不能直接用层序遍历来求高度了，这就体现出求高度和求深度的不同。
• 用迭代法，其实效率很低，因为没有很好的模拟回溯的过程，所以迭代法有很多重复的计算。虽然理论上所有的递归都可以用迭代来实现，但是有的场景难度可能比较大。例如：都知道回溯法其实就是递归，但是很少人用迭代的方式去实现回溯算法！因为对于回溯算法已经是非常复杂的递归了，如果再用迭代的话，就是自己给自己找麻烦，效率也并不一定高。
• 本题的迭代方式可以先定义一个函数，专门用来求高度。这个函数通过栈模拟的后序遍历找每一个节点的高度（其实是通过求传入节点为根节点的最大深度来求的高度）
• 然后再用栈来模拟后序遍历（递归法），遍历每一个节点的时候，再去判断左右孩子的高度是否符合。本质是前序遍历

257. 二叉树的所有路径

题目
给定一个二叉树，返回所有从根节点到叶子节点的路径。
文章
视频

递归和回溯同时存在

// 版本一
class Solution {
private:

    void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
        path.push_back(cur->val); // 中，中为什么写在这里，因为最后一个节点也要加入到path中 
        // 这才到了叶子节点
        if (cur->left == NULL && cur->right == NULL) {
            string sPath;
            for (int i = 0; i < path.size() - 1; i++) {
                sPath += to_string(path[i]);
                sPath += "->";
            }
            sPath += to_string(path[path.size() - 1]);
            result.push_back(sPath);
            return;
        }
        if (cur->left) { // 左 
            traversal(cur->left, path, result);
            path.pop_back(); // 回溯
        }
        if (cur->right) { // 右
            traversal(cur->right, path, result);
            path.pop_back(); // 回溯
        }
    }

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

404.左叶子之和

题目
文章讲解
视频

return 和result记录分开。return表示迭代到叶子，和加左叶子节点的情况并不完全重合。

class Solution {
public:
    void left(TreeNode* node, int &result)
    {
        if(node==NULL) return;
        if (node->left != NULL && node->left->left == NULL && node->left->right == NULL)
        {
            result+=node->left->val;
        }
        if(node->left) left(node->left, result);
        if(node->right) left(node->right, result);
        

    }
    int sumOfLeftLeaves(TreeNode* root) {
        int result=0;
        left(root,result);
        return result;
    }
};

retrun 和 result 重合处理：
注意对于有左叶子节点不表示迭代到头，不能return。
总的逻辑是后序‘：左右中，若左节点是左叶子则不用迭代左

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if (root == NULL) return 0;
        int  leftValue=0;   // 左
        if (root->left && !root->left->left && !root->left->right) { // 左子树就是一个左叶子的情况
            leftValue = root->left->val;
        }
        else
        leftValue = sumOfLeftLeaves(root->left);
        int rightValue = sumOfLeftLeaves(root->right);  // 右

        int sum = leftValue + rightValue;               // 中
        return sum;
    }
};

迭代法：层序遍历或前序遍历增加判断条件