挠溃症 发表于 2025-6-7 10:19:54

算法day11-二叉树(1)

目录


[*]二叉树理论基础
[*]二叉树递归遍历
[*]二叉树迭代遍历
[*]统一的迭代遍历方法
[*]小结
一、二叉树理论基础

  种类:满二叉树(除了叶子节点外,每个节点都有两个子节点)、完全二叉树(每一层都被填满,最后一层从左到右连续排列)、二叉搜索树(左子树所有节点小于根节点,右子树所有节点大于根节点)、平衡二叉搜索树(在 BST 基础上增加了高度平衡限制,使查找效率更优)。
  存储方式:

[*]1)顺序存储:用数组来存储。若父节点的数组下标是i,那么左孩子就是i*2+1,右孩子就是i*2+2。
[*]2)链式存储:
class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int val){
         this.val = val;      
    }
};

[*]3)遍历方法:

[*]深度优先遍历:前序(递归、迭代)、中序(递归、迭代)、后序(递归、迭代)
[*]广度优先遍历:层序遍历(迭代法)

二、二叉树递归遍历

   写递归函数的三个步骤:

[*]确定参数和返回值
[*]写清终止条件:加个”停止递归“的判断,不然会无限调用导致程序崩溃。
[*]写好每一层要干啥
 1. 二叉树的前序遍历(根 → 左 → 右)

  前序遍历是按照:根-左-右的顺序来访问。
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
      List<Integer> res = new ArrayList<>();
      preorder(root, res);
      return res;
    }

    public void preorder(TreeNode root, List<Integer> res){
      if(root == null){
            return;
      }
      res.add(root.val);
      preorder(root.left, res);
      preorder(root.right, res);
    }
}
//时间复杂度:O(n),其中 n 是二叉树的节点数。每一个节点恰好被遍历一次。
//空间复杂度:O(n),平均情况为O(logn),最坏情况为O(n) 2. 二叉树的中序遍历(左 → 根 → 右)

  适用于二叉搜索树,遍历结果是有序数组。
class Solution{
    public List<Integer> inorderTraversal(TreeNode root){
      List<Integer> res = new ArrayList<>();
      inorder(root, res);
      return res;
    }

    public void inorder(TreeNode root, List<Integer> res){
      if(root == null){
            return;
      }
      inorder(root.left, res);
      res.add(root.val);
      inorder(root.right, res);
    }
} 3. 二叉树的后序遍历(左 → 右 → 根)

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

    public void postorder(TreeNode root, List<Integer> res){
      if(root == null){
            return;
      }
      postorder(root.left, res);
      postorder(root.right, res);
      res.add(root.val);
    }
}三、二叉树迭代遍历

  迭代遍历主要使用 栈(Stack)来模拟递归过程,使得程序更具显式流程。
 1. 二叉树的前序遍历


[*]使用栈先处理左子树。
[*]每次访问当前节点后,将右子节点入栈,保证左子节点优先出栈。
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
      List<Integer> res = new ArrayList<>();
      if(root == null){
            return res;
      }
      Deque<TreeNode> stack = new LinkedList<>();
      TreeNode node = root;
      while(!stack.isEmpty() || node != null){
            while(node != null){
                res.add(node.val);
                stack.push(node);
                node = node.left;
            }
            node = stack.pop();
            node = node.right;
      }
      return res;
    }
}
//时间复杂度:O(n)
//空间复杂度:O(n) 2. 二叉树的中序遍历


[*]使用栈一直向左深入,直到最左叶子。
[*]退栈访问当前节点,再右移。
class Solution{
    public List<Integer> inorderTraversal(TreeNode root){
      List<Integer> res = new ArrayList<>();
      if(root == null){
            return res;
      }
      Deque<TreeNode> stack = new LinkedList<>();
      TreeNode node = root;
      while(!stack.isEmpty() || node!=null){
            while(node != null){
                stack.push(node);
                node = node.left;
            }
            node = stack.pop();
            res.add(node.val);
            node = node.right;
      }
      return res;
    }
} 3. 二叉树的后序遍历


[*]较为复杂,需记录“上一次访问的节点”prev,判断右子树是否处理过。
[*]若右子树尚未处理,需要先压栈等待右子树遍历完成。
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
      List<Integer> res = new ArrayList<>();
      if(root == null){
            return res;
      }
      Deque<TreeNode> stack = new LinkedList<>();
      TreeNode prev = null;
      while(root != null || !stack.isEmpty()){
            //先处理左边的所有节点(相对)
            while(root != null){
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();   //从栈里弹出来一个
            if(root.right == null || root.right == prev){       //如果右边没有节点或右边节点已被处理过
                res.add(root.val);
                prev = root;      //记录上一个处理过的节点
                root = null;      //这里主要是控制不用处理左边节点了
            }else{
                stack.push(root);       //处理右边的节点
                root = root.right;
            }
      }
      return res;
    }
}
//时间复杂度:O(n)
//空间复杂度:O(n) 四、二叉树层序遍历

  层序遍历采用 队列(Queue) 实现,属于 广度优先搜索(BFS)。

[*]每一层入队一次,遍历队列中所有节点,将其子节点依次加入队列。
[*]可以通过 level.size() 控制当前层级的节点数量,确保按层处理。
   这个方法特别适合:求层数、最短路径;按层收集结果;广度相关题目(如最小深度、右视图等)。
class Solution {    public List levelOrder(TreeNode root) {      List res = new ArrayList();      if(root == null){            return res;      }      Queue queue = new LinkedList();      queue.offer(root);      while(!queue.isEmpty()){            List level = new ArrayList();            int size = queue.size();            for(int i=0; i
页: [1]
查看完整版本: 算法day11-二叉树(1)