题目描述:
Given a binary tree, return the inorder traversal of its nodes' values.
For example:
Given binary tree {1,#,2,3}
,
return [1,3,2]
.
Note: Recursive solution is trivial, could you do it iteratively?
大意:中序遍历二叉树
解题思路:
1,基本的便利操作,由于是非递归算法,对于元素中栈的进出操作都必须很严谨。
2,将左边子树递归,依次入栈,取顶部元素,弹出,将递归指针赋值为top->right.这样就能将右子树入栈,实现中序遍历
代码:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector
inorderTraversal(TreeNode *root) {
vector
vec;
if(!root) return vec;
TreeNode* p=root;
stack
sta;
while(p||!sta.empty())
{
while(p)
{
sta.push(p);
p=p->left;
}
p=sta.top();
vec.push_back(p->val);
sta.pop();
p=p->right;
}
return vec;
}
};
我自己画了个图,分析了一下入栈出栈的情况,自己随便写了一个树,stack是保存的几点指针,vector是返回的容器(也代表了出栈的顺序),stack中的的数值分割虚线代表循环结束和更新指针节点信息的位置,那样比较好理解一些。
代码分析:
1,循环的条件必须为p||stack.empty().因为两者必须都为空的时候循环才能结束,递归到root节点的时候其实栈已经为空,但指针被赋值为右子树,所以就继续递归调用,直至指针指向空,并且栈内元素全部弹出
2,与其他便利方式相比,特点是边存(左子树全部入栈)边取(取顶部)边更新节点(顶部右子树)