您的位置:首页 > 其它

LeetCode 145 Binary Tree Postorder Traversal(二叉树后序遍历)

2017-05-28 09:56 465 查看
Given a binary tree, return the postorder traversal of its nodes' values.

For example:

Given binary tree 
{1,#,2,3}
,

1
\
2
/
3


return 
[3,2,1]
.

Note: Recursive solution is trivial, could you do it iteratively?

题目大意:写出二叉树非递归后序遍历的方法。

解题思路:对于任意节点N:

    1. 将N入栈

    2. 如果N没有左孩子和右孩子,或者N存在左孩子或右孩子但都已经被访问过了,那么可以直接访问N

    3. 如果N存在左孩子或右孩子且还没被访问过,那么将N的右孩子和左孩子依次入栈,然后将栈顶元素置为N,重复步骤2和3

    4. 直到N为空且栈为空,遍历结束

参考资料:http://www.cnblogs.com/dolphin0520/archive/2011/08/25/2153720.html

代码如下:

/**
* 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:
vector<int> postorderTraversal(TreeNode* root) {
if(root == nullptr) return vector<int>();
vector<int> ans;
stack<TreeNode*> stk;
TreeNode* cur;
TreeNode* pre = nullptr;
stk.push(root);

while(stk.size()){
cur = stk.top();
if((cur->left == nullptr && cur->right == nullptr) || (pre && (pre == cur->left || pre == cur->right))){
ans.push_back(cur->val);
stk.pop();
pre = cur;
}else{
if(cur->right) stk.push(cur->right);
if(cur->left) stk.push(cur->left);
}
}
return ans;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  LeetCode