另一种二叉树非递归遍历的实现

众所周知,二叉树的递归遍历十分简单,只有短短几行代码,而且三种遍历方式的代码几乎相同。而传统非递归方法的遍历无论是从代码上,还是理解上,都比递归方法要麻烦不少,尤其是后序遍历最为麻烦。为什么非递归版本的代码就不能像递归版本那样优美呢?其实,非递归版本也是可以做到的。

下面给出二叉树前中后序非递归遍历的代码,前序和后序在LeetCode上都通过了评测,如有错误,欢迎大牛指出。

#include <vector>
#include <stack>
using namespace std;
class Solution {
public:
    vector<int> vec;
    vector<int> preorderTraversal(TreeNode *root) {
        stack<pair<int, TreeNode*> > st;
        vec.clear();
        st.push(make_pair(1,root));
        while (!st.empty()) {
            pair<int, TreeNode*> p = st.top(); st.pop();
            TreeNode *node = p.second;
            if (node == NULL) continue;
            if (p.first == 0)
                vec.push_back(node->val);
            else {
                st.push(make_pair(1, node->right));
                st.push(make_pair(1, node->left));
                st.push(make_pair(0, node));
            }
        }
        return vec;
    }
    vector<int> inorderTraversal(TreeNode *root) {
        stack<pair<int, TreeNode*> > st;
        vec.clear();
        st.push(make_pair(1,root));
        while (!st.empty()) {
            pair<int, TreeNode*> p = st.top(); st.pop();
            TreeNode *node = p.second;
            if (node == NULL) continue;
            if (p.first == 0)
                vec.push_back(node->val);
            else {
                st.push(make_pair(1, node->right));
                st.push(make_pair(0, node));
                st.push(make_pair(1, node->left));
            }
        }
        return vec;
    }
    vector<int> postorderTraversal(TreeNode *root) {
        stack<pair<int, TreeNode*> > st;
        vec.clear();
        st.push(make_pair(1,root));
        while (!st.empty()) {
            pair<int, TreeNode*> p = st.top(); st.pop();
            TreeNode *node = p.second;
            if (node == NULL) continue;
            if (p.first == 0)
                vec.push_back(node->val);
            else {
                st.push(make_pair(0, node));
                st.push(make_pair(1, node->right));
                st.push(make_pair(1, node->left));
            }
        }
        return vec;
    }
};

可以看到,前中后序遍历的代码仅仅是st.push(make_pair(0, node))这行代码的位置有所变化,算法的时间复杂度和空间复杂度都是O(N)。其中pair中的第一个数表示操作类型,0代表的是打印值操作,1代表的是遍历操作。

程序的原理并不难理解,了解递归的人应该都知道,递归实际上就是一个压栈出栈的过程,程序正是模拟了这一过程。拿后序递归遍历来说,由遍历左子树、遍历右子树、打印值三个操作组成,我们可以反过来依次将打印值、遍历右子树、遍历左子树三个操作压入栈中,之后遍历左子树操作首先出栈,做完遍历左子树操作后,遍历右子树操作出栈,操作完成后打印值值操作出栈,最后程序执行完毕,这个过程和递归的过程实际上是完全一样的。


最后附上我的传统非递归版本的代码作为比较,其实前序和中序都是蛮简单的,一个在入栈时打印,一个在出栈时打印,后序要麻烦些,我们可以注意到在后序遍历中一个点如果要被打印,那么要么它是叶子节点,要么它的右孩子为空并且左孩子刚刚被打印,要么它的右孩子刚刚被打印,根据这个特点就可以写出非递归后序遍历的算法了,另外就是要注意后序遍历要在打印后才能出栈。

class Solution {
public:
    vector<int> vec;
    vector<int> preorderTraversal(TreeNode *root) {
        stack<TreeNode*> st;
        vec.clear();
        TreeNode *T = root;
        while (T || !st.empty()) {
            if (T) {
                vec.push_back(T->val);
                st.push(T);
                T = T->left;
            } else {
                T = st.top(); st.pop();
                T = T->right;
            }
        }
        return vec;
    }
    vector<int> inorderTraversal(TreeNode *root) {
        stack<TreeNode*> st;
        vec.clear();
        TreeNode *T = root;
        while (T || !st.empty()) {
            if (T) {
                st.push(T);
                T = T->left;
            } else {
                T = st.top(); st.pop();
                vec.push_back(T->val);
                T = T->right;
            }
        }
        return vec;
    }
    vector<int> postorderTraversal(TreeNode *root) {
        stack<TreeNode*> st;
        vec.clear();
        TreeNode *T = root, *last = NULL;
        while (T || !st.empty()) {
            if (T) {
                st.push(T);
                T = T->left;
            } else {
                T = st.top();
                if ((!T->left && !T->right) || (T->left == last && !T->right)
                    || (T->right == last)) {
                    vec.push_back(T->val);
                    last = T, T = NULL;
                    st.pop();
                } else {
                    T = T->right;
                }
            }
        }
        return vec;
    }
};

最后的最后,二叉树的遍历还有个空间复杂度的为O(1)的Morris算法,没有使用栈来实现,而是使用了线索二叉树的概念,也是十分巧妙的,有兴趣的童鞋可以Google一下。

OK,就到这里了。如果以上程序有什么错误,欢迎大牛批评指出。

购物商城项目采用PHP+mysql有以及html+css jq以及layer.js datatables bootstorap等插件等开发,采用了MVC模式,建立一个完善的电商系统,通过不同用户的不同需求,进行相应的调配和处理,提高对购买用户进行配置….zip项目工程资源经过严格测试可直接运行成功且功能正常的情况才上传,可轻松复刻,拿到资料包后可轻松复现出一样的项目,本人系统开发经验充足(全领域),有任何使用问题欢迎随时与我联系,我会及时为您解惑,提供帮助。 【资源内容】:包含完整源码+工程文件+说明(如有)等。答辩评审平均分达到96分,放心下载使用!可轻松复现,设计报告也可借鉴此项目,该资源内项目代码都经过测试运行成功,功能ok的情况下才上传的。 【提供帮助】:有任何使用问题欢迎随时与我联系,我会及时解答解惑,提供帮助 【附带帮助】:若还需要相关开发工具、学习资料等,我会提供帮助,提供资料,鼓励学习进步 【项目价值】:可用在相关项目设计中,皆可应用在项目、毕业设计、课程设计、期末/期中/大作业、工程实训、大创等学科竞赛比赛、初期项目立项、学习/练手等方面,可借鉴此优质项目实现复刻,设计报告也可借鉴此项目,也可基于此项目来扩展开发出更多功能 下载后请首先打开README文件(如有),项目工程可直接复现复刻,如果基础还行,也可在此程序基础上进行修改,以实现其它功能。供开源学习/技术交流/学习参考,勿用于商业用途。质量优质,放心下载使用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值