标签:中序 order lan back 前序遍历 树遍历 leetcode 递归算法 rsa
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
?? \
???2
?? /
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
后序遍历就是先遍历左子树,然后遍历右子树,然后访问根节点。
先左子树,然后右子树,最后根
vector<int> v;
vector<int> inorderTraversal(TreeNode* root) {
if(root != NULL) {
inorderTraversal(root->left);
inorderTraversal(root->right);
v.push_back(root->val);
}
return v;
}
后序遍历相对前序,中序要麻烦不少
其中一种方法是我们通过观察发现,后序遍历与前序遍历相对称。
思路: 每到一个节点 A,就应该立即访问它。 然后将左子树压入栈,再次遍历右子树。
遍历完整棵树后,结果序列逆序即可。
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> s;
vector<int> v;
TreeNode* rt = root;
while(rt || s.size()){
while(rt){
s.push(rt->left);
v.push_back(rt->val);
rt=rt->right;
}
rt=s.top();s.pop();
}
reverse(v.begin(),v.end());
return v;
}
在看了力扣上jason-2的题解后不得不说通过栈的方式确实很巧妙
思路:每到一个节点 A,因为根要最后访问,将其入栈。然后遍历左子树,遍历右子树,最后返回到 A。
但是出现一个问题,无法区分是从左子树返回,还是从右子树返回。
因此,给 A 节点附加一个标记T。在访问其右子树前,T 置为 True。之后子树返回时,当 T 为 True表示从右子树返回,否则从左子树返回。
当 T 为 false 时,表示 A 的左子树遍历完,还要访问右子树。
同时,当 T 为 True 时,表示 A 的两棵子树都遍历过了,要访问 A 了。并且在 A 访问完后,A 这棵子树都访问完成了。
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> S;
unordered_map<TreeNode*,int> done;
vector<int> v;
TreeNode* rt = root;
while(rt || S.size()){
while(rt){
S.push(rt);
rt=rt->left;
}
while(S.size() && done[S.top()]){
v.push_back(S.top()->val);
S.pop();
}
if(S.size()){
rt=S.top()->right;
done[S.top()]=1;
}
}
return v;
}
标签:中序 order lan back 前序遍历 树遍历 leetcode 递归算法 rsa
原文地址:https://www.cnblogs.com/cat-fish-/p/12787496.html