【力扣刷题】Day14——二叉树专题

文章目录

    • 二叉树的介绍
    • 二叉树的遍历
      • 1. 递归实现
      • 前序遍历
      • 中序遍历
      • 后序遍历
      • 2. 迭代实现
      • 前序遍历
      • 后序遍历
      • 中序遍历


二叉树的介绍

许多概念可以看之前的博客: 二叉树的遍历_塔塔开!!!的博客-CSDN博客_二叉树遍历

二叉树的遍历

  • 前序遍历:根左右
  • 中序遍历:左根右
  • 后序遍历:左右根
  • 层序遍历:BFS

实现方式:

  • 递归
  • 非递归(栈)

1. 递归实现

前序遍历

Code

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dfs(root, res);
        return res;
    }
    /**
        前:根左右
     */
    public static void dfs(TreeNode root, List<Integer> res){
        if(root == null){// 叶子节点
            return ;
        }
        res.add(root.val);
        dfs(root.left, res);
        dfs(root.right, res);

    }
}

中序遍历

Code

class Solution {
     List<Integer> ans = new ArrayList<>();
    public List<Integer> inorderTraversal(TreeNode root) {
        dfs(root);
        return ans;
    }
    // 中序遍历
    public void dfs(TreeNode root){
        // 结束条件
        if(root == null) return;
        dfs(root.left);
        ans.add(root.val);
        dfs(root.right);
    }
}

后序遍历

Code

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dfs(root, res);
        return res;
    }
    /**
        后:左右根
     */
    public static void dfs(TreeNode root, List<Integer> res){
        if(root == null){// 叶子节点
            return ;
        }
        dfs(root.left, res);
        dfs(root.right, res);
        res.add(root.val);

    }
}

2. 迭代实现

前序遍历

思路:用栈模拟前序遍历过程,由于是栈(先进后出)

  • 根节点先栈
  • 当栈不为空,右孩子先入栈,然后左孩子再入栈(后进先出)

栈模拟:根左右 —> 根右左

Code

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

        // 前:根左右
        Stack<TreeNode> stk = new Stack<>();
        stk.push(root);
        while(!stk.empty()){
            TreeNode node = stk.pop();
            res.add(node.val);

            // 右儿子先入栈
            if(node.right != null){
                stk.push(node.right);
            }
            // 做儿子入队
            if(node.left != null){
                stk.push(node.left);
            }
        }
        return res;
    }
}

后序遍历

上述迭代法获取前序遍历:根左右 —> 迭代过程按根右左的顺序遍历插入,最终得到前序遍历的结果(根左右

而后序遍历是左右根,它的翻转是根右左,那么我们只要想办法拿到这棵树的根左右次序,然后再将其翻转即可获得后续遍历的结果!,那么根右左的遍历次序结果我们又是如何得到的呢?—— 模仿前序遍历迭代的获取过程:

  • 根左右 —> 由入栈顺序根右左得到
  • 那么根右左 —> 由入栈顺序根左得到

Code

/**
 * 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;
 *     }
 * }
 */
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if(root == null){
            return list;
        }

        Stack<TreeNode> stk = new Stack<>();
        stk.push(root);
        // 先获取根右左的遍历次序结果
        while(!stk.empty()){
            TreeNode node = stk.pop();
            list.add(node.val);

            if(node.left != null){
                stk.push(node.left);
            }
            if(node.right != null){
                stk.push(node.right);
            }
        }
        // 翻转得到后序遍历结果
        Collections.reverse(list);
        return list;

    }
}

中序遍历

中:左根右

迭代法:

  • 定义一个指针指向根节点,当节点不为空或者栈不为空时一直循环

    • 当指针不为空时,当前节点入栈一直循环遍历左儿子,如此往复直到p指针指向空-------(模拟一直左递归的过程)
    • 当指针为空时,栈顶元素出栈,指针指向了出栈的节点,p = stk.pop(),节点值val加入ans(模拟遍历中根的过程,记录答案),然后指针p移动到当前节点的右儿子(模拟遍历中右的过程),为下一次(左根右做好准备)
  • 如此往复,直到栈为空。

Code

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        TreeNode p = root;
        Stack<TreeNode> stk = new Stack<>();
        List<Integer> ans = new ArrayList<>();

        while(p != null || !stk.empty()){
            while(p != null){// 模拟一直左递归的过程
                stk.push(p);
                p = p.left;
            }
            // p走到了空
            // 根
            p = stk.pop();
            ans.add(p.val);
            
            // 右
            p = p.right;
        }
        return ans;
    }
}

你可能感兴趣的:(代码随想录力扣刷题,leetcode,深度优先,算法)