二叉树遍历(迭代)

原创已于 2023-07-05 15:43:20 修改 · 56 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #算法 #数学建模

于 2023-06-15 17:28:46 首次发布

该文章详细展示了如何使用栈数据结构实现二叉树的前序、中序和后序遍历。代码中，`pre`函数用于前序遍历，`inOrder`函数执行中序遍历，而`postOrder`函数则处理后序遍历。每个函数都利用了一个`LinkedList`来辅助遍历过程。

private void pre(TreeNode root, List<Integer> res) {
        LinkedList<TreeNode> stack = new LinkedList<>();
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                res.add(root.val);
                stack.push(root);
                root = root.left;
            }
            root = stack.pop().right;
        }
}

private void inOrder(TreeNode root,List<Integer> list){
        LinkedList<TreeNode> stack=new LinkedList<>();
        while(root!=null || !stack.isEmpty()){
            while(root!=null){
                stack.push(root);
                root=root.left;
            }
            TreeNode parent=stack.pop();
            list.add(parent.val);
            root=parent.right;
        }
}

private void postOrder(TreeNode root, List<Integer> list){
        LinkedList<TreeNode> stack=new LinkedList<>();
        TreeNode p=null;
        while(root!=null || !stack.isEmpty()){
            while(root!=null){
                stack.push(root);
                root=root.left;
            }
            root=stack.peek();
            if(root.right == null || root.right == p){
                list.add(root.val);
                stack.pop();
                p = root;
                root = null;
            }else{
                root = root.right;
            }
        }
}