LeetCode-剑指55-I.二叉树的深度

1、深度优先遍历

我们按照深度优先遍历的顺序遍历整棵树，当我们遍历到叶子节点时比较当前的深度与最大的深度并取最大值。最终我们返回深度的最大值。

class Solution {
public:
    int res = 0, temp = 0;

    void dfs(TreeNode *root) {
        if (!root) {
            res = max(res, temp);
            return;
        }
        ++temp;
        dfs(root->left);
        dfs(root->right);
        --temp;
    }

    int maxDepth(TreeNode *root) {
        dfs(root);
        return res;
    }
};

或

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);
    }
};

2、层序遍历

我们可以使用队列迭代来层序遍历整棵树，记录的最大深度即为整棵树的深度。

class Solution  {
public:
    int maxDepth(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);
            }
        }
        return depth;
    }
};