Given a binary tree, return the postorder traversal of its nodes’ values.
Example:
Input: [1,null,2,3]
1
\
2
/
3
Output: [3,2,1]
Follow up: 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> ans;
vector<int> postorderTraversal(TreeNode* root) {
postorder(root);
return ans;
}
void postorder(TreeNode* root){
if(root == NULL)
return;
postorder(root->left);
postorder(root->right);
ans.push_back(root->val);
}
};
结果:
迭代做法: 和先序遍历和中序遍历一样,首先需要找到左下角的节点,用栈在找左下角元素的时候记录下路径上经过的节点。这段代码可以写成如下(记做代码段1):
while(p){//p初始为root节点
stk.push(p);
p = p->left;
}
经过代码段1后得到p==NULL
,这时是访问到了下面这两种情况,最近保存的是最后入栈的。图片来源
到达左下角后,假设此时栈顶元素记为plast
。左下角元素的左孩子必为NULL,但是其右孩子有两种情况,一种是其右孩子也为NULL(上图a),这样我们可以直接访问栈顶元素后让栈顶元素出栈即可。另一种情况略微复杂,其右孩子不为NULL(上图b),这时我们并不能直接访问栈顶元素,因为后序遍历的条件下,根节点应该在其右子树访问完之后访问,所以我们应该先访问栈顶元素的右子树(注意此时并没有让栈顶元素plast
出栈,因为plast还没访问),假设当前栈顶元素plast
的右子树访问完(栈可能经过一系列push和pop此时栈顶元素回到最初的plast
节点),栈顶元素还是plast
时,这时我们应该访问plast->val
,并让栈顶元素出栈,而这个操作的关键条件是plast的右孩子访问完,而我们用什么来判断当栈顶元素是第一次访问,还是栈顶元素的右子树已经访问过?我们可以设置一个变量lastVisited
指向最后被访问的节点,当栈顶元素第一次访问时有条件:lastVisited!=plast->right
,而当plast
的右子树访问过之后,也就是plast->right==lastVisited
的时候就是我们要执行上面所说的访问plast
并让其出栈的条件!
上面这些步骤完成了对一棵子树的遍历,只需要将上面的步骤循环起来就可以完成对整棵二叉树的遍历。
呼,表达能力太差了,尽力写了半天,还是写的很啰嗦。。。
talk is cheap show me the code! 如果上面说的没明白建议先看代码,之后再返回看上面这段解释,多看看应该就能理解了。
C++迭代做法代码:
/**
* 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> ans;
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> stk;
TreeNode * p = root;
TreeNode * lastVisited = NULL;//记录上一次访问过的节点
if(p == NULL)
return ans;
while(!stk.empty() || p){//每次访问完都会出栈,并且设置p=NULL
if(p){//不断往左孩子迭代,直到左孩子为空,同时路径上的节点入栈
stk.push(p);
p = p->left;
}
else{//此时p==NULL,到达了左下角节点
p = stk.top();
if(p->right == NULL || p->right == lastVisited){//右孩子为空,或者右孩子上次已经访问完
ans.push_back(p->val);//这时可以访问栈顶元素
lastVisited = p;//更新lastVisited
stk.pop();//出栈
p = NULL;//上面的操作表示p指向的节点已经访问完,因为是后序遍历,
//此时p所代表的的子树一定也访问完了,需要将指针回溯指向栈顶,访问后序元素
//设置p=NULL,可以让下一次p直接进入当前if外层的else程序段继续对栈顶元素进行访问
}
else{
p = p->right;//这时p是第一次被访问,不能直接访问其指向的值,应该先访问右子树
}
}
}
return ans;
}
};
结果:
花了好大的力气搞出迭代的做法告诉我没有递归快???excuse me??? 不明觉厉,不是很懂为什么慢了。。。