题目地址:https://leetcode.com/problems/binary-tree-postorder-traversal/#/description
题目描述:Given a binary tree, return the postorder traversal of its nodes’ values.
Note: Recursive solution is trivial, could you do it iteratively?
我的代码:
/**
* 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) {
vector<int> tree;
vector<TreeNode*> node;
if(root==NULL) return tree;
TreeNode* now;
node.push_back(root);
while(!node.empty()){
now=node.back();
tree.insert(tree.begin(),now->val);
node.pop_back();
if(now->left) node.push_back(now->left);
if(now->right) node.push_back(now->right);
}
return tree;
}
};
解题思路:
题目要求用迭代的方法后序遍历二叉树。将遍历的结果从后往前看,可知根节点在最末尾,然后是右子树,再是左子树。由此,迭代时逐步从后往前先右后左的加入子树根节点即可。
但显然的问题是,在加入右子树的根节点时,必须维护各层左子树根节点的顺序,所有采用另一个vector来保存所经过节点的子节点,即每次将该vector的最后一个节点加入后续遍历的结果最前面并在vector中删除此节点,然后将该节点的左右子节点加入vector中,注意,必须先左后右,如此在下一步使用时才会先右后左,符合我们之前分析的顺序。这样当保存完整个右子树后,自然会转到对应的左子树。(这一点通过观察叶节点就很容易明白。)
如此,我们在程序中只遍历了这个树一遍,复杂度是O(n)。