最全二叉树的遍历方式总结

本文囊括了二叉树遍历的所有方式,基本所有二叉树的题目中,都要基于本文的实现,进行改造和升级。
推荐扔进收藏吃灰

最全二叉树的遍历方式总结

一、前序遍历

1、递归实现

不加说明后面树节点的定义方式都为:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

这里参看 leetcode 144. 二叉树的前序遍历

题目描述:给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        preOrder(root, res);
        return res;
    }

    public void preOrder(TreeNode node, List res) {
        if (node == null) return ;
        res.add(node.val);	// 访问
        preOrder(node.left, res);	// 左
        preOrder(node.right, res);	// 右
    }
}

2、迭代实现

方法一: 模仿递归实现的思路

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
          	// 因为利用栈进行遍历的时候是从栈顶部进行弹出
          	// 所以要想实现先左后右的访问必须先将右节点入栈
            if (node != null) {
                res.add(node.val);	// 访问
                stack.push(node.right);	// 右
                stack.push(node.left);	// 左
            }
        }
        return res;
    }
}

方法二: 栈的回溯的思路

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Deque<TreeNode> stk = new LinkedList<>();
      	// 判空
        while (root != null || !stk.isEmpty()) {
          	// 先遍历至没有左节点,过程中不断压栈
            while (root != null) {
                // 访问节点
                res.add(root.val);
                stk.push(root);
                root = root.left;
            }
          	// 没有左节点后,弹出
        		// 此时弹出的为最左节点
            root = stk.pop();
          	
          	// 无左节点后压入右节点,继续循环
            root = root.right;
        }
        return res;
    }
}

二、中序遍历

1、递归实现

leetcode 94.二叉树的中序遍历

题目描述:给你二叉树的根节点 root ,返回它节点值的 中序 遍历。

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        inOrder(root, res);
        return res;
    }

    public void inOrder(TreeNode node, List res) {
        if (node == null) return;
        inOrder(node.left, res);
        res.add(node.val);
        inOrder(node.right, res);
        
    }
}

2、迭代实现

这里要注意区别于前序遍历,我们不能简单的再次调转前序遍历中的顺序就实现了。

详细见代码注释

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Deque<TreeNode> stk = new LinkedList<>();
      	// 判空
        while (root != null || !stk.isEmpty()) {
          	// 先遍历至没有左节点,过程中不断压栈
            while (root != null) {
                stk.push(root);
                root = root.left;
            }
          	// 没有左节点后,弹出
        		// 此时弹出的为最左节点
            root = stk.pop();
          	// 访问节点
            res.add(root.val);
          	// 无左节点后压入右节点,继续循环: 左->左->左...访问->
            root = root.right;
        }
        return res;
    }
}

三、后序遍历

1、递归实现

leetcode 145. 二叉树的后序遍历

题目描述:给你二叉树的根节点 root ,返回它节点值的 后序 遍历。

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        postOrder(root, res);
        return res;
    }

    public void postOrder(TreeNode node, List res) {
        if (node == null) return;
        postOrder(node.left, res);
        postOrder(node.right, res);
        res.add(node.val);
    }
}

2、迭代实现

后序遍历较于前序和中序要稍稍复杂一点点,我们需要维护一个指针来判断当前节点是否为 三次访问过的节点(包括叶子节点)

这里定义一个概念方便我们去理解,在访问一个节点时,需要经历这种步骤如下:

第一次抵达->遍历左孩子->第二次抵达->遍历右孩子->第三次抵达->弹出(不再访问)

此时我们定义这个不在访问到的节点为无效节点

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();							
        Deque<TreeNode> stack = new LinkedList<TreeNode>();				
        TreeNode prev = null;																			// new
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
          	// 弹出叶子节点或无效节点的父节点
            root = stack.pop();
          	// 判断是否将要成为无效节点
          	// 右孩子为空或者为prev说明,右孩子是无效节点,即将弹栈,节点将成为无效节点
            if (root.right == null || root.right == prev) {
              	// 成为无效节点前进行访问
                res.add(root.val);
              	// 访问后标记无效节点
                prev = root;
              	// 置空,表示无效节点,并准备下次循环弹栈
                root = null;
            } else {
                stack.push(root);
                root = root.right;
            }
        }
        return res;
    }
}

四、迭代法统一的代码风格(*)

下面的方法摘自代码随想录大佬。

1、前序

class Solution {
  
  public List<Integer> preorderTraversal(TreeNode root) {
      List<Integer> result = new LinkedList<>();
      Stack<TreeNode> st = new Stack<>();
      if (root != null) st.push(root);
      while (!st.empty()) {
          TreeNode node = st.peek();
          if (node != null) {
              st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
              if (node.right!=null) st.push(node.right);  // 添加右节点(空节点不入栈)
              if (node.left!=null) st.push(node.left);    // 添加左节点(空节点不入栈)
              st.push(node);                          // 添加中节点
              st.push(null); // 中节点访问过,但是还没有处理,加入空节点做为标记。
              
          } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
              st.pop();           // 将空节点弹出
              node = st.peek();    // 重新取出栈中元素
              st.pop();
              result.add(node.val); // 加入到结果集
          }
      }
      return result;
  }
}


2、中序

class Solution {
  public List<Integer> inorderTraversal(TreeNode root) {
  	    List<Integer> result = new LinkedList<>();
      Stack<TreeNode> st = new Stack<>();
      if (root != null) st.push(root);
      while (!st.empty()) {
          TreeNode node = st.peek();
          if (node != null) {
              st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
              if (node.right!=null) st.push(node.right);  // 添加右节点(空节点不入栈)
              st.push(node);                          // 添加中节点
              st.push(null); // 中节点访问过,但是还没有处理,加入空节点做为标记。

              if (node.left!=null) st.push(node.left);    // 添加左节点(空节点不入栈)
          } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
              st.pop();           // 将空节点弹出
              node = st.peek();    // 重新取出栈中元素
              st.pop();
              result.add(node.val); // 加入到结果集
          }
      }
      return result;
  }
}

3、后序

class Solution {

 public List<Integer> postorderTraversal(TreeNode root) {
      List<Integer> result = new LinkedList<>();
      Stack<TreeNode> st = new Stack<>();
      if (root != null) st.push(root);
      while (!st.empty()) {
          TreeNode node = st.peek();
          if (node != null) {
              st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
              st.push(node);                          // 添加中节点
              st.push(null); // 中节点访问过,但是还没有处理,加入空节点做为标记。
              if (node.right!=null) st.push(node.right);  // 添加右节点(空节点不入栈)
              if (node.left!=null) st.push(node.left);    // 添加左节点(空节点不入栈)         
                             
          } else { // 只有遇到空节点的时候,才将下一个节点放进结果集
              st.pop();           // 将空节点弹出
              node = st.peek();    // 重新取出栈中元素
              st.pop();
              result.add(node.val); // 加入到结果集
          }
      }
      return result;
 }
}


五、层序遍历(一通万通的板子)

1、基础的层序遍历

  • leetcode 102. 二叉树的层序遍历

在这里插入图片描述

题目描述:

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

这里我们写一个层序遍历的板子

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Deque<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        queue.offer(root);
      	// 队列不为空
        while (!queue.isEmpty()) {
            List<Integer> level = new ArrayList<>();
          	// 当前队列的容量即为当前层数节点的容量
            int currentSize = queue.size();
          	// 遍历当前层
          	// 循环结束后,队列的 前面部分为当前层的节点,此时已经全部弹出
          	// 然后在循环遍历时在队列后面已经加入了下一层的全部节点
            for (int i = 0; i < currentSize; i ++) {
              	// 弹出遍历过的节点
                TreeNode node = queue.poll();
              	// 将弹出的节点值进行记录
                level.add(node.val);
              	// 将左孩子加入队列
                if (node.left != null) {
                    queue.offer(node.left);
                }
              	// 将右孩子加入队列
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(level);
        }
        return res;
    }
}

2、从下到上的层序遍历

  • leetcode 107. 二叉树的层序遍历 II
    在这里插入图片描述

解题思路:

根据上面已经写过的层序遍历,我们只需要在遍历的时候将下一层的结果插入上一层的结果之前就可以了,详情看代码变动的部分。

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int currentSize = queue.size();
            List<Integer> level = new ArrayList<>();
            for (int i = 0; i < currentSize; i ++) {
                TreeNode node = queue.poll();
                level.add(node.val);
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(0, level); // 只需要在这里调整一下插入的位置就可以
        }
        return res;
    }
}

3、二叉树的右视图

  • leetcode 199. 二叉树的右视图

在这里插入图片描述

解题思路:

依然是二叉树的层序遍历,只不过我们这次只需要存储每一层中最后一个节点的值就可以了。

详细看代码标注的位置

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int currentSize = queue.size();
            TreeNode node = root;		// 这里要将变量拿出来以便循环外访问, 并设置第一层的时候默认为root节点
            for (int i = 0; i < currentSize; i ++) {
                node = queue.poll();	// 没有变化,依然是遍历queue的时候存储,在循环结束后为当前层的最后一个节点
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(node.val);	// 将最后一个节点的值加入结果集中
        }
        return res;
    }
}

4、二叉树的层平均值

  • leetcode 637. 二叉树的层平均值

在这里插入图片描述

因为太简单了,直接把我们的板子粘过来。

class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
        Deque<TreeNode> queue = new LinkedList<>();
        List<Double> res = new ArrayList<>();
        if (root == null) return res;
        queue.offer(root);
      	// 队列不为空
        while (!queue.isEmpty()) {
            List<Integer> level = new ArrayList<>();
          	// 当前队列的容量即为当前层数节点的容量
            int currentSize = queue.size();
            double sum = 0;
          	// 遍历当前层
          	// 循环结束后,队列的 前面部分为当前层的节点,此时已经全部弹出
          	// 然后在循环遍历时在队列后面已经加入了下一层的全部节点
            for (int i = 0; i < currentSize; i ++) {
              	// 弹出遍历过的节点
                TreeNode node = queue.poll();
              	// 将弹出的节点值进行记录
                sum += node.val;
              	// 将左孩子加入队列
                if (node.left != null) {
                    queue.offer(node.left);
                }
              	// 将右孩子加入队列
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(sum / currentSize);
        }
        return res;
    }
}

5、n叉树的层序遍历

  • leetcode 429. n叉树的层序遍历

在这里插入图片描述

解题思路:

和二叉树的层序遍历是一个思路,只不过在遍历孩子节点的时候是通过List来访问的。

详细见下面代码标注的部分。

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        Deque<Node> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int currentSize = queue.size();
            List<Integer> level = new ArrayList<>();
            for (int i = 0; i < currentSize; i ++) {
                Node node = queue.poll();
                level.add(node.val);
              	// 这里在访问孩子节点的时候通过 list for的形式进行遍历即可
                for (Node subNode : node.children) {
                    queue.offer(subNode);
                }
            }
            res.add(level);
        }
        return res;
    }
}
/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/

6、在每层树找最大值

  • leetcode 515.在每个树行中找最大值

在这里插入图片描述

解题思路:

依然是太简单,粘板子

class Solution {
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int currentSize = queue.size();
            int max = Integer.MIN_VALUE; // 这里用来记录最大值
            for (int i = 0; i < currentSize; i ++) {
                TreeNode node = queue.poll();
                max = Math.max(max, node.val);
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(max); // 将每一层记录的最大值记录进结果集
        }
        return res;
    }
}

7、填充每个节点的下一个右侧指针

在这里插入图片描述
在这里插入图片描述

上板子,随便改改:

class Solution {
    public Node connect(Node root) {
        if (root == null) return null;
        Deque<Node> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int currentSize = queue.size();
            for (int i = 0; i < currentSize; i ++) {
                Node node = queue.poll();
              	// 只需要在这里变更一下是否是最后一个节点,不是就链接到下一个节点
                if (i < currentSize - 1) {
                    node.next = queue.peek();
                }
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
        }
        return root;
    }
}
  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值