(C++)算法记录:通过迭代遍历求二叉树最大深度
最近重新刷算法题,有道题需要求二叉树最大深度,发现对迭代遍历方法有点犯迷糊,为此通过本篇文章记录下来,方便以后复习时看。
1. 问题:
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
题目给出了二叉树节点的定义,如下所示:
/**
* 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) {}
* };
*/
2. 解决方案:
因为对使用 迭代遍历方法有疑惑,因此在这里主要记录迭代方法的一些细节问题。
为实现迭代法遍历二叉树,在遍历的过程中,使用 栈来存储节点(队列也可以)。
下面是迭代法后序遍历一颗二叉树并计算最大深度的代码:
std::stack<TreeNode*> st;
if(root)
st.push(root);
int depth = 0;
int result = 0;
while(!st.empty()){
TreeNode* node = st.top();
if(node){
st.pop();
st.push(node);
st.push(NULL);
depth++;
if(node->right)
st.push(node->right);
if(node->left)
st.push(node->left);
}
else{
st.pop();
TreeNode* node = st.top();
st.pop();
depth--;
}
result = result > depth ? result: depth;
}
return result;
下面简单分析一下代码:
个人认为上一段代码中depth之所以自减处理,是因为处理这个节点后,深度应该减一,回到上一个深度的状态,如果要继续处理下一层深度节点,则depth再加1,这样以此类推,处理完所有节点之后,depth的值应该为0(代表这棵树已经遍历并且处理完了),所以result的作用现在看起来也比较明朗了: 在遍历过程中记录下来depth能达到的最大值,如果depth增大,就更新result,如果depth开始减小(回溯到上层节点),这个result就不作修改了。
参考资料:微信公众号“代码随想录"