题目链接
题意
给定一颗二叉树,求该二叉树的最大深度。
题解
本题属于基础题,可以通过 DFS
和 BFS
来求解。
题解1 DFS
如果该结点为NULL,则以该结点为根的子树深度为0. 如果该结点非NULL,则求其左右子树的深度,取两者的最大值并+1.
此处DFS
为递归版本,当然为了防止栈溢出,也可以用栈来写非递归版本,或者写使用BFS
来解题。
C++代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int maxDepth(TreeNode* root) {
return root == NULL ? 0 : max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
};
题解2 BFS
BFS
需要通过队列来实现。大致思想是,每一轮都取出当前层的全部结点,并将这些结点的所有孩子加入队列,直至二叉树的全部结点都被取出(即队列为空)。用一个变量来记录轮数,也就是二叉树的最大深度。
C++代码
class Solution {
public:
int maxDepth(TreeNode* root) {
if (root== NULL)
return 0;
queue<TreeNode*> q;
q.push(root);
int res = 0, tmp;
TreeNode *node;
while(!q.empty()){
tmp = q.size(); // 每一轮取出当前行全部结点(即队列里的全部结点)
for(int i = 0; i < tmp; i++){
node = q.front();
q.pop();
if(node->left != NULL)
q.push(node->left);
if(node->right != NULL)
q.push(node->right);
}
res++; // 记录轮数
}
return res;
}
};