C++二叉树前序中序后序迭代遍历
Posted 小小鱼的代码基地
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了C++二叉树前序中序后序迭代遍历相关的知识，希望对你有一定的参考价值。
前序遍历很简单，先把当前节点的值保存到ret中，然后再把它的右孩子节点、左孩子节点分别压入stack中。注意是先右后左。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> ret;
stack<TreeNode*> s;
s.push(root);
while(!s.empty()){
TreeNode* cur = s.top();
s.pop();
if(cur != NULL){
ret.push_back(cur->val);
s.push(cur->right);
s.push(cur->left);
}
}
return ret;
}
};
中序遍历，稍微复杂点：
根据中序遍历的顺序，对于任一结点，优先访问其左孩子，而左孩子结点又可以看做一根结点，然后继续访问其左孩子结点，直到遇到左孩子结点为空的结点才进行访问，然后按照相同的规则访问其 右子树。因此处理过程如下：
对任一结点p
1.若其左孩子不为空，则将p入栈并将p的左孩子置为当前的p，然后对当前结点p再进行相同的处理；
2.若其左孩子为空，则取栈顶元素并进行出栈操作，访问该栈顶结点，然后将栈顶结点的右孩子置为当前的p
3.直到p为null且栈为空则结束遍历
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> ret;
stack<TreeNode*> s;
TreeNode* p = root;
while(p || !s.empty()){
while(p != NULL){
s.push(p);
p = p -> left;
}
TreeNode* cur = s.top();
s.pop();
ret.push_back(cur->val);
p = cur -> right;
}
return ret;
}
};
后序遍历也较复杂，但是看见网上有个比较好记住的办法：按照与前序相似的方法（前序压栈的顺序是先右后左，这里是先左后右），先得到一个结果，然后对结果倒序一下，就可以了。
class Solution {
public:
//按照和前序相似的思想，然后再倒一次顺序即可
vector<int> postorderTraversal(TreeNode* root) {
vector<int> ret;
stack<TreeNode*> s;
s.push(root);
while(!s.empty()){
TreeNode* cur = s.top();
s.pop();
if(cur != NULL){
ret.push_back(cur->val);
s.push(cur->left); //前序遍历时，是先右后左的。这里先左后右。
s.push(cur->right);
}
}
reverse(ret.begin(), ret.end());
return ret;
}
};
以上是关于C++二叉树前序中序后序迭代遍历的主要内容，如果未能解决你的问题，请参考以下文章