Day17：代码随想录算法训练营第17天| 110.平衡二叉树 | 257.二叉树的所有路径

110.平衡二叉树

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

题解：

本题采用后序遍历，依旧是递归三部曲。

1. 确定参数和返回值

int getHight(TreeNode* node)

2.确定返回值

if (node == NULL){
 return 0;
}

3.明确单层递归逻辑,分别求出其左右子树的高度，然后如果差值小于等于1，则返回当前二叉树的高度，否则返回-1，表示已经不是二叉平衡树了

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;

总结C++代码如下：

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）​​​​​​

题解：

本题依旧可以使用递归来求解，但在递归的同时还要使用回溯法，具体过程如下图所示

 （本图出自：代码随想录 (programmercarl.com)）

 1.确定递归参数以及返回值，要传入根节点，记录每一条路径的path和存放结果集的result，这里递归不需要返回值。

void traversal(TreeNode* cur, vector<int>*&path,vector<string>& result)

 2.确定递归终止条件

if (cur->left == NULL && cur->right == NULL) { // 遇到叶子节点
    string sPath;
    for (int i = 0; i < path.size() - 1; i++) { // 将path里记录的路径转为string格式
        sPath += to_string(path[i]);
        sPath += "->";
    }
    sPath += to_string(path[path.size() - 1]); // 记录最后一个节点（叶子节点）
    result.push_back(sPath); // 收集一个路径
    return;
}

3.确定单层递归逻辑

if (cur->left) {
     traversal(cur->left, path, result);
     path.pop_back();
}
if (cur->right) {
     traversal(cur->right, path, result);
      path.pop_back();
}

总结C++代码如下：

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;
    }
};