【代码随想录第15天| 110.平衡二叉树，257. 二叉树的所有路径，404.左叶子之和，222.完全二叉树的节点个数】

110.平衡二叉树

思路

• 比较高度一般采用后序遍历。

1. 递归函数的参数：当前传入节点， 返回值：以当前传入节点为根节点的树的高度。
2. 终止条件：遇到空节点，返回0，表示当前节点的根节点的树高度为0.
3. 单层递归逻辑：判断左子树、右子树高度的差值。

代码随想录

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) return -1;
        result = 1 + max(leftHeight, rightHeight);
        return result;
    }

    bool isBalanced(TreeNode* root) {
        if (root == NULL) return true;
        return getHeight(root) == -1 ? false : true;
    }
};

257. 二叉树的所有路径

思路

• 是一个带有明显回溯特点的题目，因此考虑用栈来实现回溯功能

1. 递归函数的参数：路径vector<int> path，存放结果集的vector<string> result。不需要返回值，因此递归函数设计为void。
2. 递归终止条件：找到叶子节点。即该节点不为空，但是其左右孩子都为空。
3. 终止处理逻辑：把路径path添加到结果集result中
4. 确定单层递归逻辑：前序遍历。中：把路径上的节点值添加进path，左：递归并pop_back()，右：递归并pop_back()。

代码随想录

class Solution {
public:
    // 确定递归函数的参数以及返回值
    void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
        // 中
        path.push_back(cur->val);       
        // 终止条件
        if (cur->left == NULL && cur->right == NULL) {
            string string_path;
            for (int i = 0; i < path.size() - 1; i++) {
                string_path += to_string(path[i]);
                string_path += "->";
            }
            string_path += to_string(path[path.size() - 1]);
            result.push_back(string_path);
            return;
        }
        // 左
        if (cur->left) {
            traversal(cur->left, path, result);
            path.pop_back();
        }
        // 右
        if (cur->right) {
            traversal(cur->right, path, result);
            path.pop_back();
        }    
    }

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

404.左叶子之和

思路

• 左叶子：叶子节点 && 是其根节点的左孩子

1. 递归函数参数：根节点， 返回值：数值和int
2. 确定终止条件：遍历到空节点
3. 确定单层递归逻辑：遇到左叶子节点时，记录数值，并递归求左子树左叶子之和和右子树左叶子之和，相加便是整个树的左叶子之和

代码随想录

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); // 左
        int rightValue = sumOfLeftLeaves(root->right); // 右
        // 判断：当前节点的左节点不空，当前节点的左节点的左右节点都为空
        // 此时记录下数值
        if (root->left && !root->left->left && !root->left->right) {
            leftValue = root->left->val;
        }
        int sum = leftValue + rightValue; // 中
        return sum;
    }
};

222.完全二叉树的节点个数

思路

• 普通二叉树的思路就是直接查节点。

1. 递归函数的参数：当前节点， 递归函数的返回值：以当前节点为根节点的二叉树的节点数量int。
2. 确定终止条件：空节点返回0，表示节点数为0。
3. 单层递归逻辑：前序遍历，递归查左节点，递归查右节点，中的处理为做加法

class Solution {
private:
    int getNodesNum(TreeNode* cur) {
        if (cur == NULL) return 0;
        int leftNum = getNodesNum(cur->left);      // 左
        int rightNum = getNodesNum(cur->right);    // 右
        int treeNum = leftNum + rightNum + 1;      // 中
        return treeNum;
    }
public:
    int countNodes(TreeNode* root) {
        return getNodesNum(root);
    }
};

迭代法

class Solution {
public:
    int countNodes(TreeNode* root) {
        queue<TreeNode*> que;
        int result = 0;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode *node = que.front();
                que.pop();
                result ++;
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return result;
    }
};

代码随想录

• 完全二叉树只有两种情况，情况一：满二叉树，情况二：最后一层叶子节点不满
• 如果整个树不是满二叉树，就递归其左右孩子，直到遇到满二叉树为止。
• 如何判断是否为满二叉树：递归向左遍历的深度等于递归向右遍历的深度。

class Solution {
public:
    int countNodes(TreeNode* root) {
        if (root == NULL)
            return 0;
        int left_dep = 0;
        int right_dep = 0;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        // 终止条件：找到完全二叉树
        while (left) {
            left_dep++;
            left = left->left;
        }
        while (right) {
            right_dep++;
            right = right->right;
        }
        if (left_dep == right_dep) {
            return (2 << left_dep) - 1;
        }
        // 若两个dep不相等，则返回和
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};