数据结构以学带练day16——树的最大深度、二叉树的最小深度、完全二叉树的节点个数

133 阅读5分钟

题目

104.二叉树的最大深度

image.png

深度遍历法(后序&递归)

本题可以使用前序(中左右),也可以使用后序遍历(左右中),使用前序求的就是深度,使用后序求的是高度。

  • 二叉树节点的深度:指从根节点到该节点的最长简单路径边的条数或者节点数(取决于深度从0开始还是从1开始)
  • 二叉树节点的高度:指从该节点到叶子节点的最长简单路径边的条数后者节点数(取决于高度从0开始还是从1开始)

而根节点的高度就是二叉树的最大深度,所以本题中我们通过后序求的根节点高度来求的二叉树最大深度。

  1. 确定递归函数的参数和返回值:参数就是传入树的根节点,返回就返回这棵树的深度,所以返回值为int类型。
  2. 确定终止条件:如果为空节点的话,就返回0,表示高度为0。
  3. 确定单层递归的逻辑:先求它的左子树的深度,再求右子树的深度,最后取左右深度最大的数值 再+1 (加1是因为算上当前中间节点)就是目前节点为根节点的树的深度。
class solution {
public:
    int getdepth(treenode* node) {
        if (node == NULL) return 0;
        int leftdepth = getdepth(node->left);       // 左
        int rightdepth = getdepth(node->right);     // 右
        int depth = 1 + max(leftdepth, rightdepth); // 中
        return depth;
    }
    int maxdepth(treenode* root) {
        return getdepth(root);
    }
};
//精简版
class solution {
public:
    int maxdepth(treenode* root) {
        if (root == null) return 0;
        return 1 + max(maxdepth(root->left), maxdepth(root->right));
    }
};

扩展:559.n叉树的最大深度

递归法

class solution {
public:
    int maxdepth(node* root) {
        if (root == 0) return 0;
        int depth = 0;
        for (int i = 0; i < root->children.size(); i++) {
            depth = max (depth, maxdepth(root->children[i]));
        }
        return depth + 1;
    }
};

层序遍历法

class Solution {
public:
    int maxDepth(Node* root) {
        //定义队列
        queue<Node*> q;
        int depth = 0;
        //若树不为空
        if(root!=nullptr) q.push(root);
        //队列不为空,遍历每层
        while(q.empty()==false){
            int size = q.size();//记录每层节点数
            for(int i = 0 ;i< size;i++){
                Node* node = q.front();
                q.pop();
                //如果子节点不为空,则加入到队列
                for(int j=0;j<node->children.size();j++){
                    q.push(node->children[j]);
                }
            }
            depth++;//遍历完一层
        }
        return depth;
    }

111. 二叉树的最小深度

image.png

递归法

  1. 确定递归函数的参数和返回值:参数为要传入的二叉树根节点,返回的是int类型的深度。
  2. 确定终止条件:终止条件也是遇到空节点返回0,表示当前节点的高度为0。
  3. 确定单层递归的逻辑:
    • 如果左子树为空,右子树不为空,说明最小深度是 1 + 右子树的深度。
    • 如果右子树为空,左子树不为空,最小深度是 1 + 左子树的深度。
    • 如果左右子树都不为空,返回左右子树深度最小值 + 1 。
class Solution {
public:
    int getDepth(TreeNode* node) {
        if (node == NULL) return 0;
        int leftDepth = getDepth(node->left);           // 左
        int rightDepth = getDepth(node->right);         // 右
                                                        // 中
        // 当一个左子树为空,右不为空,这时并不是最低点
        if (node->left == NULL && node->right != NULL) { 
            return 1 + rightDepth;
        }   
        // 当一个右子树为空,左不为空,这时并不是最低点
        if (node->left != NULL && node->right == NULL) { 
            return 1 + leftDepth;
        }
        int result = 1 + min(leftDepth, rightDepth);
        return result;
    }

    int minDepth(TreeNode* root) {
        return getDepth(root);
    }
};

层序遍历法(Day15)

class Solution {
public:

    int minDepth(TreeNode* root) {
        if (root == NULL) return 0;
        int depth = 0;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            int size = que.size();
            depth++; // 记录最小深度
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
                if (!node->left && !node->right) { // 当左右孩子都为空的时候,说明是最低点的一层了,退出
                    return depth;
                }
            }
        }
        return depth;
    }
};

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

image.png

递归法

  1. 确定递归函数的参数和返回值:参数就是传入树的根节点,返回就返回以该节点为根节点二叉树的节点数量,所以返回值为int类型。
  2. 确定终止条件:如果为空节点的话,就返回0,表示节点数为0。
  3. 确定单层递归的逻辑:先求它的左子树的节点数量,再求右子树的节点数量,最后取总和再加一 (加1是因为算上当前中间节点)就是目前节点为根节点的节点数量。
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*> q;
        int nodenum = 0;//记录节点总数
        if(root!=nullptr) q.push(root);
        while(q.empty()==false){
            int size = q.size();//记录每层节点数
            //遍历每层
            for(int i=0;i<size;i++){
                TreeNode* node = q.front();
                q.pop();
                if(node->left) q.push(node->left);
                if(node->right) q.push(node->right);
            }
            nodenum+=size;
        }
        return nodenum;
    }
};

根据完全二叉树性质分析求解(比较复杂)

完全二叉树只有两种情况:

  • 情况一:就是满二叉树,
  • 情况二:最后一层叶子节点没有满。

对于情况一,可以直接用 2树深度12^{树深度} - 1 来计算,注意这里根节点深度为1。

对于情况二,分别递归左孩子,和右孩子,递归到某一深度一定会有左孩子或者右孩子为满二叉树,然后依然可以按照情况1来计算。如下图:

image.png

所以,如果整个树不是满二叉树,就递归其左右孩子,直到遇到满二叉树为止,用公式计算这个子树(满二叉树)的节点数量

在完全二叉树中,如果递归向左遍历的深度等于递归向右遍历的深度,那说明就是满二叉树。如图:

image.png

在完全二叉树中,如果递归向左遍历的深度不等于递归向右遍历的深度,则说明不是满二叉树,如图:

image.png

class Solution {
public:
    int countNodes(TreeNode* root) {
        /*================2、递归终止条件=================================*/
        if (root == nullptr) return 0;
        
        // 开始根据左深度和右深度是否相同来判断该子树是不是满二叉树
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        int leftDepth = 0, rightDepth = 0; // 这里初始为0是有目的的,为了下面求指数方便
        while (left) {  // 求左子树深度
            left = left->left;
            leftDepth++;
        }
        while (right) { // 求右子树深度
            right = right->right;
            rightDepth++;
        }
        //如果是满二叉树
        if (leftDepth == rightDepth) {
            return (2 << leftDepth) - 1; // 注意(2<<1) 相当于2^2,所以leftDepth初始为0
        }
        /*================2、递归终止条件=================================*/
        
        int leftTreeNum = countNodes(root->left);       // 左
        int rightTreeNum = countNodes(root->right);     // 右
        int result = leftTreeNum + rightTreeNum + 1;    // 中
        return result
    }
};