模拟系统栈实现二叉树前序遍历

最新推荐文章于 2024-08-17 23:55:10 发布

原创最新推荐文章于 2024-08-17 23:55:10 发布 · 354 阅读

1 ·

CC 4.0 BY-SA版权

算法和刷题专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用Java实现二叉树前序遍历的方法，通过自定义命令类来控制遍历流程，利用栈结构实现了递归思想的非递归写法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

import java.util.*;
public class Main {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) {
            val = x;
        }
    }

	//系统指令
    public class Command{
        String type;    //指令分两种，一种是访问 go，一种是执行print
        TreeNode node;
        public Command(String type,TreeNode node){
            this.type = type;
            this.node = node;
        }
    }


    public static void main(String[] args) {

    }

    public List<Integer> preorderTraversal(TreeNode root) {
        Stack<Command> sysStack = new Stack<Command>();	//系统栈
        List<Integer> list = new ArrayList<Integer>();
        if (root != null){
            sysStack.push(new Command("go",root));
        }
        while (!sysStack.isEmpty()){
            Command cmd = sysStack.pop();
            if (cmd.type == "go"){
            	//由于栈是后进先出，所以这里入栈的顺序和遍历的顺序相反
            	//只要改变下面三部分的执行顺序，就可以变为中序遍历和前序遍历
                if (cmd.node.right != null){
                    sysStack.push(new Command("go",cmd.node.right));
                }
                if (cmd.node.left != null){
                    sysStack.push(new Command("go",cmd.node.left));
                }
                sysStack.push(new Command("print",cmd.node));
            }else {
                assert cmd.type == "print";
                list.add(cmd.node.val);
            }
        }
        return list;
    }
}