算法训练营 第十七天

算法训练营 第十七天

二叉树 day04~

110.平衡二叉树

递归法
判断平衡二叉树的关键其实就是计算高度，判断左子树和右子树的高度是否相差不大于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 {
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.左叶子之和

这道题我的逻辑是，后序遍历，如果发现了左叶子结点（在左子树那边发现了叶子结点）就收集。

class Solution {
public:
    bool digui(TreeNode* root, int& res) {
        if (root == nullptr)
            return false;
        if (root->left == nullptr&&root->right==nullptr)
            return true;

        if (digui(root->left, res)) {
            res+=root->left->val;
        }
        digui(root->right, res);
        return false;
        
    }
    int sumOfLeftLeaves(TreeNode* root) {
        int res = 0;
        digui(root, res);
        return res;
        

    }
};

代码随想录中的逻辑是各统计左右子树的左叶子结点值，然后在根节点处汇总。

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if (root == NULL) return 0;
        if (root->left == NULL && root->right== NULL) return 0;

        int leftValue = sumOfLeftLeaves(root->left);    // 左
        if (root->left && !root->left->left && !root->left->right) { // 左子树就是一个左叶子的情况
            leftValue = root->left->val;
        }
        int rightValue = sumOfLeftLeaves(root->right);  // 右

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