题目
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
递归解法
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;
}
};