二叉树的前序、中序、后序的迭代算法实现

public static List<Integer> preOrder(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }

        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode get = stack.pop();
            list.add(get.value);
            //利用堆栈先进后出,先加入右子树
            if (get.right != null) {
                stack.push(get.right);
            }
            //再加入左子树
            if (get.left != null) {
                stack.push(get.left);
            }
        }

        return list;
    }
 //二叉树的中序遍历的迭代实现,利用 栈 模拟 递归过程
    public static List<Integer> infixOrder(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        //两个while语句
        TreeNode p = root;
        Stack<TreeNode> stack = new Stack<>();
        while (p != null || !stack.isEmpty()) {
            while (p != null) {//加入当前结点和当前结点的左子树
                stack.push(p);
                p = p.left;
            }

            p = stack.pop();
            list.add(p.value);//访问
            p = p.right;//访问右子树

        }

        return list;
    }
 //前序遍历为:根左右
    //后序遍历为:左右根
    //把前序遍历的左子树和右子树调换顺序,按前序遍历在逆序,就得出后序遍历的迭代
    public List<Integer> postOrderByPreOrder(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }

        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        TreeNode p = null;

        while (!stack.isEmpty()) {
            p = stack.pop();
            list.add(p.value);//访问根节点

            //调换左右的顺序
            //先进后出
            //左子树先进栈
            if (p.left != null) {
                stack.push(p.left);
            }
            //右子树再进栈,右子树先出栈
            if (p.right != null) {
                stack.push(p.right);
            }
        }

        return list;//结果反转就是后序遍历
    }
//后序遍历的迭代算法,一个节点出栈与否,跟它是否有右子树有关,
    // 没有右子树,则出栈;有右子树,但已访问过,也出栈;有右子树且未访问,访问右子树
    public List<Integer> postOrder(TreeNode root) {
        if (root == null) {//空二叉树
            return list;
        }

        TreeNode p = root;
        TreeNode last = null;//标志上一次访问的结点
        Stack<TreeNode> s = new Stack<>();

        while (p != null || !s.isEmpty()) {
            while (p != null) {
                s.push(p);//加入此结点和其左孩子
                p = p.left;
            }

            p = s.peek();
            //如果栈顶结点没有右子树或者右子树已经被访问过,则访问改节点
            if (p.right == null || (last != null && last == p.right)) {
                s.pop();
                list.add(p.value);
                last = p;//更新下一次的上一次访问结点
                p = null;
            } else {//有右子树且未访问,访问右子树
                p = p.right;
                last = null;
            }
        }

        return list;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树遍历方式有四种:前序遍历中序遍历后序遍历和层次遍历。下面分别介绍这四种遍历方式的算法思想和实现方法: 1. 前序遍历 前序遍历算法思想是先访问根节点,然后按照左子树、右子树的顺序递归遍历每个子树。具体实现过程可以采用递归方式或者利用栈来实现迭代遍历。 递归方式实现前序遍历: ``` void preOrderTraversal(TreeNode* root) { if (root != NULL) { cout << root->val << " "; //访问根节点 preOrderTraversal(root->left); //递归遍历左子树 preOrderTraversal(root->right); //递归遍历右子树 } } ``` 利用栈实现前序遍历: ``` void preOrderTraversal(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> st; st.push(root); while (!st.empty()) { TreeNode* node = st.top(); st.pop(); cout << node->val << " "; //访问当前节点 if (node->right != NULL) st.push(node->right); //右子树入栈 if (node->left != NULL) st.push(node->left); //左子树入栈 } } ``` 2. 中序遍历 中序遍历算法思想是先递归遍历左子树,然后访问根节点,最后递归遍历右子树。具体实现过程可以采用递归方式或者利用栈来实现迭代遍历。 递归方式实现中序遍历: ``` void inOrderTraversal(TreeNode* root) { if (root != NULL) { inOrderTraversal(root->left); //递归遍历左子树 cout << root->val << " "; //访问根节点 inOrderTraversal(root->right); //递归遍历右子树 } } ``` 利用栈实现中序遍历: ``` void inOrderTraversal(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> st; TreeNode* node = root; while (node != NULL || !st.empty()) { while (node != NULL) { st.push(node); node = node->left; } node = st.top(); st.pop(); cout << node->val << " "; //访问当前节点 node = node->right; } } ``` 3. 后序遍历 后序遍历算法思想是先递归遍历左子树,然后递归遍历右子树,最后访问根节点。具体实现过程可以采用递归方式或者利用栈来实现迭代遍历。 递归方式实现后序遍历: ``` void postOrderTraversal(TreeNode* root) { if (root != NULL) { postOrderTraversal(root->left); //递归遍历左子树 postOrderTraversal(root->right); //递归遍历右子树 cout << root->val << " "; //访问根节点 } } ``` 利用栈实现后序遍历: ``` void postOrderTraversal(TreeNode* root) { if (root == NULL) return; stack<TreeNode*> st; TreeNode* node = root; TreeNode* lastVisit = NULL; while (node != NULL || !st.empty()) { while (node != NULL) { st.push(node); node = node->left; } node = st.top(); if (node->right == NULL || node->right == lastVisit) { st.pop(); cout << node->val << " "; //访问当前节点 lastVisit = node; node = NULL; } else { node = node->right; } } } ``` 4. 层次遍历 层次遍历算法思想是按照从上到下、从左到右的顺序依次访问每个节点。具体实现过程可以采用队列来实现。 ``` void levelOrderTraversal(TreeNode* root) { if (root == NULL) return; queue<TreeNode*> q; q.push(root); while (!q.empty()) { int size = q.size(); for (int i = 0; i < size; i++) { TreeNode* node = q.front(); q.pop(); cout << node->val << " "; //访问当前节点 if (node->left != NULL) q.push(node->left); //左子树入队 if (node->right != NULL) q.push(node->right); //右子树入队 } } } ``` 以上是二叉树实现前序中序后序、层次遍历算法思想和实现方法。其中,递归方式实现简单易懂,但可能会导致递归调用栈溢出的问题;利用栈或队列实现迭代遍历方式更加稳定,但需要额外的空间来存储栈或队列。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值