二叉树前,中,后序的非递归实现

10 篇文章 0 订阅

力扣题的链接在下面

前序链接
中序链接
后序链接

二叉树的前,中,序用递归很简单我这里就不写了,主要来看非递归的实现
前序的非递归

思路:用栈的后进先出来控制结点,还是比较简单的

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
    	//用栈来存放树的结点
        stack<TreeNode*> st;
        //遍历的结果放在数组返回
        vector<int> ret;
        //用cur遍历树
        TreeNode* cur = root;
        //结束条件是当cur为空,并且栈里没有结点了
        while(cur || !st.empty())
        {
        	//循环每次都是去左节点当cur为空这个子树的左节点就访问完了
            while(cur)
            {
            	//前序是先记录数据再左,再右
            	//保存要返回的数据
                ret.push_back(cur->val);
                //每次访问的结点都保持,入栈
                st.push(cur);
                //向左树走
                cur = cur->left;
            }
			//走到这里这里就说明了,当前子树左结点访问完了,保存这个结点,去访问它的右结点
            TreeNode* top = st.top();
            //保存了就可以出栈了
            st.pop();
            //这个结点的值已经保存了,并且这个结点的左节点已经访问过了,访问右树
            cur = top->right;
        }
        return ret;
    }
};

中序

思路: 中序跟前序的思路差不多,只是要先找到最左结点

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> ret;
        TreeNode* cur = root;
        while(cur || !st.empty())
        {
        	//cur指向谁就表示开始访问它的左结点
        	//左路结点入栈
            while(cur)
            {
                st.push(cur);
                cur = cur->left;
            }
			//化为子问题去访问它的左节点
            TreeNode* top = st.top();
            st.pop();
            //取出的结点,说明它的右结点已经访问过了
            ret.push_back(top->val);
            cur = top->right;
        }

        return ret;
    }
};

后序

后序比前,中序都要难一点,比较抽象,先看波代码

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> ret;
        TreeNode* cur;
        TreeNode* prev;
        while(cur || st.empty())
        {
            whlie(cur)
            {
            	//左路结点入栈
                st.push(cur);
                cur = cur->left;
            }
            
            TreeNode* top = st.top();
            if(top->right == nullptr ||top->right == prev)
            {
                st.pop();
                ret.push_back(top->val);
                prev = top;
            }
            else
            {
                cur = top->right;
            }
        }
        return ret;
   

没有注释不好理解
思路:前面还是跟上面一样,还是先访问树的左结点,入栈,后序是先左子树,右子树,根,
情况1就是结点的right为空,这个时候就可以访问这个结点
情况2:right不为空,那么这个时候我们就需要去迭代它的右树,化为子问题,要注意的是当right不为空,我们还没有迭代它的右树,我们是不可以访问这个结点需要先去访问它的右结点,这个时候我们可以用一个前驱指针指向我们出栈的,就是上一个访问的结点,如果它的右树的根是上一个访问的那么就说明了右树访问过了,可以访问该结点了 (top->right = prev)这个判段条件,右树为空就判空,不是空,看上一个访问的是不是这个右树的根来判断是不是访问过,后序遍历是比较难的,挺抽象的不容易理解,建议画画图跟着代码

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
以下是C语言实现二叉树后序遍历非递归完整代码: ```c #include <stdio.h> #include <stdlib.h> #define MAXSIZE 100 typedef struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; } TreeNode; typedef struct Stack { TreeNode *data[MAXSIZE]; int top; } Stack; void initStack(Stack *s) { s->top = -1; } int isEmpty(Stack *s) { return s->top == -1; } int isFull(Stack *s) { return s->top == MAXSIZE - 1; } void push(Stack *s, TreeNode *node) { if (isFull(s)) { printf("Stack is full.\n"); return; } s->data[++s->top] = node; } TreeNode *pop(Stack *s) { if (isEmpty(s)) { printf("Stack is empty.\n"); return NULL; } return s->data[s->top--]; } TreeNode *createTree() { int val; scanf("%d", &val); if (val == -1) { return NULL; } TreeNode *root = (TreeNode *)malloc(sizeof(TreeNode)); root->val = val; root->left = createTree(); root->right = createTree(); return root; } void preOrder(TreeNode *root) { if (root == NULL) { return; } Stack s; initStack(&s); push(&s, root); while (!isEmpty(&s)) { TreeNode *node = pop(&s); printf("%d ", node->val); if (node->right != NULL) { push(&s, node->right); } if (node->left != NULL) { push(&s, node->left); } } } void inOrder(TreeNode *root) { if (root == NULL) { return; } Stack s; initStack(&s); TreeNode *node = root; while (node != NULL || !isEmpty(&s)) { while (node != NULL) { push(&s, node); node = node->left; } node = pop(&s); printf("%d ", node->val); node = node->right; } } void postOrder(TreeNode *root) { if (root == NULL) { return; } Stack s1, s2; initStack(&s1); initStack(&s2); push(&s1, root); while (!isEmpty(&s1)) { TreeNode *node = pop(&s1); push(&s2, node); if (node->left != NULL) { push(&s1, node->left); } if (node->right != NULL) { push(&s1, node->right); } } while (!isEmpty(&s2)) { TreeNode *node = pop(&s2); printf("%d ", node->val); } } int main() { TreeNode *root = createTree(); printf("Preorder traversal: "); preOrder(root); printf("\nInorder traversal: "); inOrder(root); printf("\nPostorder traversal: "); postOrder(root); return 0; } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

学习新算法

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值