好久没写, 忙于实习以及晚上刷刷题, 这次来写一下平时很多人都很畏惧的二叉树非递归遍历, 说实话, 我自己也挺怕递归转成非递归的, 因为有些转起来比较简单, 有些转起来就很困难了, 这次我们来试试二叉树的非递归遍历, 难度倒是不算大.
前序遍历的非递归写法
前序遍历的非递归写法要简单不少, 我们来简单捋一下思路, 我们需要先遍历根节点, 再遍历左子树的节点, 再遍历右子树的节点, 完成遍历. 另外, 一般来说, 由递归转化为非递归, 我们都需要用到栈, 这里也不例外. 下面直接上代码, 原理都在代码里面进行解释.
- void preOrder(TreeNode* root)
- {
- if(root == nullptr)
- {
- return;
- }
- stack<int> s;
- auto p = root;
- while(!s.empty() || p)
- {
- // 直接访问树的最左节点, 因为是先序遍历所以每次访问先打印节点的值
- while(p)
- {
- s.push(p);
- cout<<p.val<<endl;
- p = p->left;
- }
- if(!s.empty())
- {
- auto node = s.top();
- s.pop();
- p = node->right;
- }
- }
- }
中序遍历的非递归写法
在三种非遍历算法中, 中序遍历应该是最简单的了. 实现代码 q 其实比较好玩的是和前序遍历基本一样 (笑), 只是改了一下输出的位置.
- void inOrder(TreeNode* root)
- {
- if(root == nullptr)
- {
- return;
- }
- stack<TreeNode*> s;
- auto p = root;
- while(!s.empty() || p)
- {
- // 到达树的最左节点, 由于是中序遍历, 所以先不打印节点的值
- while(p)
- {
- s.push(p);
- p = p->left;
- }
- if(!s.empty())
- {
- auto node = s.top();
- s.pop();
- cout<<node.val<<endl;
- p = node->right;
- }
- }
- }
后序遍历的非递归写法
这个应该是三种非递归的写法中最难的了, 因为我们需要判断当我们从子树返回的时候判断是从左子树返回还是从右子树返回的, 所以我们需要一个 prev 指针来保存我们上一次访问的指针, 其余的部分其实和之前也没有差很多.
- void postOrder(TreeNode* root)
- {
- if(root == nullptr)
- {
- return;
- }
- stack<TreeNode*> s;
- auto p = node;
- TreeNode* prev = nullptr;
- while(!s.empty() || p)
- {
- while(p)
- {
- s.push(p);
- p = p->left;
- }
- if(!s.empty())
- {
- auto node = s.top();
- s.pop();
- if(!node->right || node->right == prev)
- {
- cout<<node->val;
- prev = node;
- }
- else
- {
- s.push(node);
- p = node->right;
- }
- }
- }
- }
总结: 其实上面的代码风格十分统一, 我个人觉得理解起来还是很简单的, 主要核心还是理解遍历的顺序问题, 我们每次都先访问到树的最左边, 然后不断的压栈, 思路其实并不难理解.
来源: http://www.jianshu.com/p/2c1aad1448f7