代码随想录算法训练营第16天 |第六章 二叉树part03

学习目标：

• 104.二叉树的最大深度
• 559.n叉树的最大深度
• 111.二叉树的最小深度
• 222.完全二叉树的节点个数

---

学习内容：

104.二叉树的最大深度

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int getDepth(TreeNode* cur){
        if(cur==nullptr) return 0;
        int leftDepth = getDepth(cur->left);
        int rightDepth = getDepth(cur->right);
        int depth = 1+max(leftDepth,rightDepth);
        return depth;
    }
    int maxDepth(TreeNode* root) {
        return getDepth(root);
    }
};

错误以及注意事项

• cur==nullptr表示没有子结点。

559. n叉树的最大深度

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    int getDepth(Node* cur){
        if(cur==0) return 0;
        int depth=0;
        for(int i=0;i<cur->children.size();i++)
        {
            depth = max(depth,getDepth(cur->children[i]));
        }
        return 1+ depth;
    }
    int maxDepth(Node* root) {
        return getDepth(root);
    }
};

错误以及注意事项

• cur==0 表示没有子结点。

111. 二叉树的最小深度

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution
{
public:
    int getDepth(TreeNode *cur)
    {
        if (cur == nullptr)
            return 0;
        int leftDepth = getDepth(cur->left);
        int rightDepth = getDepth(cur->right);

        if (cur->left != nullptr && cur->right == nullptr)
        {
            return 1 + leftDepth;
        }
        if (cur->right != nullptr && cur->left == nullptr)
        {
            return 1 + rightDepth;
        }
        int result = 1 + min(leftDepth, rightDepth);
        return result;
    }
    int minDepth(TreeNode *root)
    {
        return getDepth(root);
    }
};

错误以及注意事项

• 注意的是，最小深度的定义是从根节点到最近叶子节点的最短路径上的节点数量，所以求二叉树的最小深度和求二叉树的最大深度的差别主要在于处理左右孩子不为空的逻辑。
  

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

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution
{
public:
    int getSumNode(TreeNode *cur)
    {
        if (cur == nullptr)
            return 0;
        int leftSum = getSumNode(cur->left);
        int rightSum = getSumNode(cur->right);
        int sum = 1 + leftSum + rightSum;
        return sum;
    }
    int countNodes(TreeNode *root)
    {
        return getSumNode(root);
    }
};

错误以及注意事项

• 这一题乍一看感觉很奇怪，这不就是直接输出给的数组的大小就可以了吗？再仔细一看，给的是根节点。

---

学习时间：

2023.12.15 19:00-20:25
时间有点着急，所以只看了递归法，最近忙的有点不知道在干什么了。
今天把数据集搞好了，但是运行是多GPU跑还没成功，明天研究一下。