二叉树的几种遍历方式(附 LeetCode 水题)

二叉树的几种遍历方式(附 LeetCode 水题)_第1张图片

正文之前

闲得无聊去刷 LeetCode 的时候做了一点深搜和广搜的题,但是树的遍历方式还没有写过总结,今天刚好总结一下:

  1. 深度优先搜索
    • 前序遍历
    • 中序遍历
    • 后序遍历
  2. 广度优先搜索

正文

1. 节点的定义

这里定义树的节点的方式就按照 LeetCode 上给出的吧:

class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
}

接下来给出用例:

二叉树的几种遍历方式(附 LeetCode 水题)_第2张图片
        TreeNode node10 = new TreeNode(10);
        TreeNode node7 = new TreeNode(7);
        TreeNode node8 = new TreeNode(8);
        TreeNode node1 = new TreeNode(1);
        TreeNode node3 = new TreeNode(3);
        TreeNode node0 = new TreeNode(0);
        TreeNode node2 = new TreeNode(2);
        TreeNode node4 = new TreeNode(4);
        TreeNode node6 = new TreeNode(6);
        node10.left = node7;
        node10.right = node8;
        node7.left = node1;
        node7.right = node3;
        node3.left = node0;
        node3.right = node2;
        node8.left = node4;
        node8.right = node6;

在遍历时用 ArrayList 来存放根节点的值,

2. 深度优先搜索

分别有三种方式,前序先访问根节点,中序第二个访问根节点,后序最后访问根节点,这里写给出三种方法,然后统一测试一下:

    //前序遍历
    public void preOrderTraversal(TreeNode node, ArrayList list){
        if (node == null){
            return;
        }
        //先访问根节点
        list.add(node.val);
        //然后是左子节点和右子节点
        preOrderTraversal(node.left, list);
        preOrderTraversal(node.right, list);
    }

    //中序遍历
    public void inOrderTraversal(TreeNode node, ArrayList list){
        if (node == null){
            return;
        }
        //先访问左子节点,然后是根节点,最后是右子节点
        inOrderTraversal(node.left, list);
        list.add(node.val);
        inOrderTraversal(node.right, list);
    }

    //后序遍历
    public void postOrderTraversal(TreeNode node, ArrayList list){
        if (node == null){
            return;
        }
        postOrderTraversal(node.left, list);
        postOrderTraversal(node.right, list);
        list.add(node.val);
    }

根据我们定义的用例,三种遍历方式的输出分别是这样的:

二叉树的几种遍历方式(附 LeetCode 水题)_第3张图片
        ArrayList list = new ArrayList<>();

        System.out.print("前序遍历:");
        s.preOrderTraversal(node10, list);
        for (Integer i: list) {
            System.out.print(i + " ");
        }
        list.clear();
        System.out.println();

        System.out.print("中序遍历:");
        s.inOrderTraversal(node10, list);
        for (Integer i: list) {
            System.out.print(i + " ");
        }
        list.clear();
        System.out.println();

        System.out.print("后序遍历:");
        s.postOrderTraversal(node10, list);
        for (Integer i: list) {
            System.out.print(i + " ");
        }

得到输出:

前序遍历:10 7 1 3 0 2 8 4 6
中序遍历:1 7 0 3 2 10 4 8 6
后序遍历:1 0 2 3 7 4 6 8 10

3. 广度优先搜索

广搜的方式就是一层一层的访问节点,又叫做层次遍历,用队列来实现:

    public void BFS(TreeNode node, ArrayList list){
        Queue queue = new LinkedList<>();
        if (node != null){
            //根节点作为第一层,先加入
            queue.add(node);
            while (!queue.isEmpty()){
                TreeNode cur = queue.poll();
                list.add(cur.val);
                //依次加入左右子节点
                if (cur.left != null){
                    queue.offer(cur.left);
                }
                if (cur.right != null){
                    queue.offer(cur.right);
                }
            }
        }
    }

遍历的结果是:

层次遍历:10 7 8 1 3 4 6 0 2 

我用图来走一遍流程:

二叉树的几种遍历方式(附 LeetCode 水题)_第4张图片

每次出队列一个节点,然后把它的子节点(如果存在)加入到队列,循环往复,直到队列为空

4. LeetCode 水题

既然介绍了树的遍历,就随便拿两个简单的题来做个例子吧:

257. Binary Tree Paths

求二叉树的所有路径,这一题可能和遍历有一点区别(因为是路径,根节点会出现多次),不过想法是一样的,用的前序,先访问根节点,然后是左右子节点:

    public List binaryTreePaths(TreeNode root) {
        String s = "";
        List list = new ArrayList<>();
        if (root != null){
            DFS(s, root, list);
        }
        
        return list;
        
    }
    public void DFS(String s, TreeNode node, List list){
        if (node.left == null && node.right == null){
            list.add(s + node.val);
        }
        if (node.left != null){
            DFS(s + node.val + "->", node.left, list);
        }
        if (node.right != null){
            DFS(s + node.val + "->", node.right, list);
        }
    }

513. Find Bottom Left Tree Value

求二叉树最底层的最左边节点,用的层次遍历,只不过这次是从右往左,最后一个出队列的就是目标节点了:

    public int findBottomLeftValue(TreeNode root) {
        Queue queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()){
            root = queue.poll();
            if (root.right != null){
                queue.add(root.right);
            }if (root.left != null){
                queue.add(root.left);
            }
        }
        return root.val;
    }

关于树的遍历就到这里了

你可能感兴趣的:(二叉树的几种遍历方式(附 LeetCode 水题))