欢迎访问宙启技术站
智能推送

Java中如何实现二叉树的前序遍历函数

发布时间:2023-06-22 15:08:05

二叉树是一种非常常见的数据结构,常用于数据的存储和查找。二叉树有三种遍历方式:前序遍历、中序遍历和后序遍历。前序遍历是指先访问根节点,再访问左子树,最后访问右子树的遍历方式。在Java中实现二叉树的前序遍历函数,可以使用递归或者栈的方式。

使用递归实现二叉树的前序遍历:

递归是指函数自己调用自己,可以很方便的实现二叉树的前序遍历。前序遍历的方式是:先访问根节点,再访问左节点,再访问右节点。递归实现的思路是,先访问根节点,再递归访问左子树,最后递归访问右子树。

代码实现如下:

public void preorderTraversal(TreeNode root) {
    if (root != null) {
        System.out.print(root.val + " ");  // 访问根节点
        preorderTraversal(root.left);     // 递归遍历左子树
        preorderTraversal(root.right);    // 递归遍历右子树
    }
}

其中,TreeNode是二叉树的节点类,具体实现如下:

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

使用栈实现二叉树的前序遍历:

使用栈实现二叉树的前序遍历需要注意的是,根节点要先入栈,然后依次访问每个节点的左子树,当左子树遍历完成后再遍历右子树。因此,在入栈的时候,需要先将右子树压入栈中,这样才能保证遍历顺序正确。

代码实现如下:

public void preorderTraversal(TreeNode root) {
    if (root == null) return;

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

    while (!stack.isEmpty()) {
        TreeNode node = stack.pop();
        System.out.print(node.val + " ");

        if (node.right != null) {
            stack.push(node.right);
        }

        if (node.left != null) {
            stack.push(node.left);
        }
    }
}

以上就是在Java中实现二叉树的前序遍历函数的两种方式。通过递归或者栈的方式实现,可以灵活的遍历二叉树,从而满足不同的需要。