原题:
代码:
Given a binary tree, return the postorder traversal of its nodes' values.
For example:
Given binary tree [1,null,2,3]
,
1 \ 2 / 3
return [3,2,1]
.
Note: Recursive solution is trivial, could you do it iteratively?
解题方法:
用了两种方法来实现:
方法一:
- 先将根结点压入栈,然后定义一个辅助结点head
- while循环的条件是栈不为空,在循环中,首先将栈顶结点取出来,如果栈顶结点没有左右子结点,或者其左子结点是head,或者其右子结点是head的情况下。我们将栈顶结点值加入结果res中,并将栈顶元素移出栈,然后将head指向栈顶元素;
- 否则的话就看如果右子结点不为空,将其加入栈,再看左子结点不为空的话,就加入栈,
这里先右后左的顺序是因为栈的后入先出的特点,可以使得左子结点先被处理。head每次就是指向前一个处理过并且加入结果res的结点,那么如果栈顶结点的左子结点或者右子结点是head的话,说明其子结点已经加入结果res了,那么就可以处理当前结点了。
方法二:
由于后序遍历的顺序是左-右-根,而先序遍历的顺序是根-左-右, 在先序遍历的方法上做些小改动,使其遍历顺序变为根-右-左,最后将结果翻转一下,就是左-右-根。
代码:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> res;
stack<TreeNode*> s;
if (root)
s.push(root);
TreeNode* head = root;
while(!s.empty()){
TreeNode* node = s.top();
if ( (!node->left && !node->right) || node->left == head || node->right == head ){
s.pop();
res.push_back(node->val);
head = node;
continue;
}
if (node->right)
s.push(node->right);
if (node->left)
s.push(node->left);
}
return res;
}
vector<int> postorderTraversal1(TreeNode* root) {
vector<int> res;
stack<TreeNode*> s;
if (root)
s.push(root);
while(!s.empty()){
TreeNode* node = s.top();
s.pop();
res.push_back(node->val);
if (node->left)
s.push(node->left);
if (node->right)
s.push(node->right);
}
reverse(res.begin(), res.end());
return res;
}