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

正文之前

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

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

正文

1. 节点的定义

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

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

接下来给出用例:

        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<Integer> list){
        if (node == null){
            return;
        }
        //先访问根节点
        list.add(node.val);
        //然后是左子节点和右子节点
        preOrderTraversal(node.left, list);
        preOrderTraversal(node.right, list);
    }

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

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

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

        ArrayList<Integer> 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<Integer> list){
        Queue<TreeNode> 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 

我用图来走一遍流程:

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

4. LeetCode 水题

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

257. Binary Tree Paths

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

    public List<String> binaryTreePaths(TreeNode root) {
        String s = "";
        List<String> list = new ArrayList<>();
        if (root != null){
            DFS(s, root, list);
        }
        
        return list;
        
    }
    public void DFS(String s, TreeNode node, List<String> 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<TreeNode> 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;
    }

关于树的遍历就到这里了

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,108评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,699评论 1 296
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,812评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,236评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,583评论 3 288
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,739评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,957评论 2 315
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,704评论 0 204
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,447评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,643评论 2 249
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,133评论 1 261
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,486评论 3 256
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,151评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,108评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,889评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,782评论 2 277
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,681评论 2 272

推荐阅读更多精彩内容