【力扣hot100题】(036)二叉树的最大深度
用递归的方法没什么好说的,说一下学到的广度优先bfs。
广度优先就是构造一个栈,每次完栈底压入栈顶元素的左右子树,然后删除栈顶元素。
这里学到的是每次记录栈元素,这样就不会不同层的元素搞混。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int maxDepth(TreeNode* root) {
if(root==nullptr) return 0;
queue<TreeNode*> node;
int result=0;
node.push(root);
while(!node.empty()){
int size=node.size();
result++;
while(size--){
if(node.front()->left) node.push(node.front()->left);
if(node.front()->right) node.push(node.front()->right);
node.pop();
}
}
return result;
}
};
其实这个方法在空间复杂度上不如dfs(时间复杂度一样都是o(n)),而且也更复杂。