每日一練（27）：二叉樹的深度

加班猿發表於2022-03-04
二叉樹

title: 每日一練（27）：二叉樹的深度

categories:[劍指offer]

tags:[每日一練]

date: 2022/02/26

每日一練（27）：二叉樹的深度

輸入一棵二叉樹的根節點，求該樹的深度。從根節點到葉節點依次經過的節點（含根、葉節點）形成樹的一條路徑，最長路徑的長度為樹的深度。

例如：

給定二叉樹 [3,9,20,null,null,15,7]，

    3
   / \
  9  20
    /  \
   15   7

返回它的最大深度 3 。

提示：

節點總數 <= 10000

來源：力扣（LeetCode）

連結：https://leetcode-cn.com/probl...

方法一：遞迴

後續遍歷：

按照遞迴三部曲，來看看如何來寫。

1.確定遞迴函式的引數和返回值：引數就是傳入樹的根節點，返回就返回這棵樹的深度，所以返回值為int型別。
程式碼如下：

int getDepth(TreeNode* node)

2.確定終止條件：如果為空節點的話，就返回0，表示高度為0。
程式碼如下：

if (node == NULL) return 0;

3.確定單層遞迴的邏輯：先求它的左子樹的深度，再求右子樹的深度，最後取左右深度最大的數值 再+1 （加1是因為算上當前中間節點）就是目前節點為根節點的樹的深度。

int leftDepth = getDepth(node->left);       // 左
int rightDepth = getDepth(node->right);     // 右
int depth = 1 + max(leftDepth, rightDepth); // 中
return depth;
//1簡化前
int getDepth(TreeNode *root) {
    if (root == 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);
}
//2簡化後
int maxDepth(TreeNode* root) {
    return (root == NULL) ? 0 : 1 + max(maxDepth(root->left), maxDepth(root->right));
}

方法二：迭代法(BFS)

使用迭代法的話，使用層序遍歷是最為合適的，因為最大的深度就是二叉樹的層數，和層序遍歷的方式極其吻合

int maxDepth(TreeNode* root) {
    if (root == NULL) {
        return 0;
    }
    int depth = 0;
    queue<TreeNode*> que;
    que.push(root);
    while (!que.empty()) {
        depth++;
        int size = que.size();
        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);
            }
        }
    }
    return depth;
}

相關文章