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

题目

输入一棵二叉树的根节点，求该树的深度。从根节点到叶节点依次经过的节点（含根、叶节点）形成树的一条路径，最长路径的长度为树的深度。

 

递归解法

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (!root)
            return 0;
        return max(maxDepth(root->right), maxDepth(root->left)) + 1;
    }
};

 

非递归解法

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (!root)
            return 0;

        int ans = 0;
        queue<pair<TreeNode*, int>> que;
        que.push({root, 1});
        while (!que.empty()) {
            auto p = que.front();
            que.pop();
            if (p.first->right) {
                que.push({p.first->right, p.second + 1});
            }
            if (p.first->left) {
                que.push({p.first->left, p.second + 1});
            }
            if (!p.first->right && !p.first->left) {
                ans = max(ans, p.second);
            }
        }

        return ans;
    }
};

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (!root)
            return 0;

        int depth = 0;
        queue<TreeNode*> que;
        que.push(root);
        while (!que.empty()) {
            ++depth;

            queue<TreeNode*> tmp;
            while (!que.empty()) {
                auto node = que.front();
                que.pop();
                if (node->right)
                    tmp.push(node->right);
                if(node->left)
                    tmp.push(node->left);
            }
            que = tmp;
        }

        return depth;
    }
};