> 文档中心 > 《Java数据结构》二叉树的这些基本操作你真的理解了吗

《Java数据结构》二叉树的这些基本操作你真的理解了吗


✅作者简介:大家好,我是小鱼儿,大家可以叫我鱼儿

📒博客首页:是小鱼儿哈

🔥系列专栏:Java数据结构

🌻每日一句:人的一生,可以有所作为的时机只有一次,那就是现在!

💖博主也在学习阶段,如发现问题请告知,非常感谢💖

目录

一、二叉树的快速构建:

二、二叉树的遍历

🌰前序遍历 

🌰中序遍历

🌰后续遍历

🌰层序遍历

三、其他一些常见操作

🍑获取树中结点的个数

🍑获取叶子结点的个数 

🍑获取第k层结点的个数 

🍑获取二叉树的高度 

🍑检测value结点是否存在 

 四、完整的代码

 五、测试代码


我们上一篇博客聊了聊二叉树的概念和性质,那么今天就让我们看看二叉树都有那些常见的操作吧!

💖要对二叉树进行操作,那么首先我们要构建一个二叉树,但因为我们刚学二叉树,并且我们本篇博客的重点是学习二叉树的基本操作。

所以此处我们先手动快速创建一棵简单的二叉树,快速进入二叉树操作学习,等我们二叉树结构了解的差不多时,我们反过头再来研究二叉树真正的创建方式
 

一、二叉树的快速构建:

// 树类public class MyBinaryTree {    // 结点类    class TreeNode { public char val;  // 储存该结点的数值 public TreeNode left; //左孩子的引用,即该结点的左子结点 public TreeNode right; //右孩子的引用 // 三种对树的结点的构造方法(每一种方法所传参数都是不同的) TreeNode() { } // 其实我们接下来用到的都是下面这一种构造方法,因为接下来我们只是在构造方法中给该结点赋值,至于该结点的左右子结点,我们是手动给他们建立联系的 TreeNode(char val) {     this.val = val; } TreeNode(char val, TreeNode left, TreeNode right) {     this.val = val;     this.left = left;     this.right = right; } @Override  // 重写toString方法,一般接下来能够之间打印树的结点 public String toString() {     return "TreeNode{" +      "val=" + val +      '}'; }    }    // 其实这不是二叉树真正的构建方式,但我们的目的是快速构建一个二叉树,学习的二叉树的一些方法(这是我们的侧重点)    // 至于真正的构建方法,我们之后的博客会讲到,本篇博客我们重点了解二叉树所对应的各种独特的方法    public TreeNode createBinaryTree(){ TreeNode node1 = new TreeNode('A'); TreeNode node2 = new TreeNode('B'); TreeNode node3 = new TreeNode('C'); TreeNode node4 = new TreeNode('D'); TreeNode node5 = new TreeNode('E'); TreeNode node6 = new TreeNode('F'); TreeNode root = node1;  // 我们采用了穷举法来快速的构建一棵二叉树 node1.left = node2; node1.right = node3; node2.left = node4; node2.right = node5; node3.left = node6; return root;     // 返回我们构建树的根结点    }}

二、二叉树的遍历

学习二叉树结构,最简单的方式就是遍历。所谓遍历(Traversal)是指沿着某条搜索路线,依次对树中每个结点均做一次且仅做一次访问。访问结点所做的操作依赖于具体的应用问题(比如:打印节点内容、节点内容加1)

遍历是二叉树上最重要的操作之一,是二叉树上进行其它运算之基础
 

 

大家想一下:在遍历二叉树时,如果没有进行某种约定,每个人都按照自己的方式遍历,得出的结果就比较混乱

🔔但如果按照某种规则进行约定,则每个人对于同一棵树的遍历结果肯定是相同的。如果N代表根节点,L代表根节点的左子树,R代表根节点的右子树,则根据遍历根节点的先后次序有以下遍历方式:

  • NLR:前序遍历(Preorder Traversal 亦称先序遍历)——访问根结点--->根的左子树--->根的右子树
  •  LNR:中序遍历(Inorder Traversal)——根的左子树--->根节点--->根的右子树。 
  • LRN:后序遍历(Postorder Traversal)——根的左子树--->根的右子树--->根节点
     
// 前序遍历void preOrder(TreeNode root);// 中序遍历void inOrder(TreeNode root);// 后序遍历void postOrde(TreeNode root);

🌰前序遍历 

比如这样一棵树

   

 它的前序遍历顺序就是:1->2->3->4->5->6

他所对应的代码就是:

// 前序遍历——》根->左子树->右子树    public void preOrder(TreeNode root) { if (root == null) return; System.out.print(root.val + " "); postOrder(root.left); postOrder(root.right);    }

好家伙,这短短几行就完成了,这递归还真是简短呀! 

📝不过他的核心思想就是 先处理左右子树所对应的根结点,再处理左子树,最后再处理右子树

为啥呢,来看看这个图,你就知道了

   

🌰中序遍历

📝核心思想是一样的,只是顺序变了,先处理左子树,再处理左右子树所对应的根结点,最后再处理右子树

 代码如下:

// 中序遍历——》左子树—>根->右子树    public void inOrder(TreeNode root) { if (root == null) return; inOrder(root.left); System.out.print(root.val + " "); inOrder(root.right);    }

🌰后续遍历

这个道理是一样的😁

// 后续遍历——》左子树->右子树—>根    public void postOrder(TreeNode root) { if (root == null) return; postOrder(root.left); postOrder(root.right); System.out.print(root.val + " ");    }

🌰层序遍历

📝设二叉树的根节点所在层数为1,层序遍历就是从所在二叉树的根节点出发,首先访问第一层的树根节点,然后从左到右访问第2层的节点,接着是第三层的节点,以此类推,自上而下,自左至右逐层访问树的结点的过程就是层序遍历。
 

 //层序遍历    public void levelOrder(TreeNode root) { if(root == null) return; Queue queue = new LinkedList(); // 我们需要借助一个队列, queue.offer(root); // 先把根结点放到队列中,根结点相当于是第一层 while (!queue.isEmpty()) {      // 借助tmp来把当前结点的左右子树给放到队列中     // 注意随着队列元素的不断弹出,tmp是在动态变化的,所以才能变量完——整棵二叉树     TreeNode tmp = queue.poll();  // 因为这一层的元素入队列的时候,是通过上一层元素的对左右子树操作来实现的,入队列是一层一层入的     System.out.print(tmp.val + " ");     if (tmp.left != null) {  queue.offer(tmp.left);  // 把当前从队列中弹出的结点的左子结点放到队列中     }     if (tmp.right != null) {  queue.offer(tmp.right); // 右子结点     } }    }

   // 力扣oj题目——层序遍历,因为要求不同所以我们的具体做法有些不一样,但基本的思想相同    public List<List> levelOrder1(TreeNode root) { List<List> ret = new ArrayList(); if (root == null) return ret; // Queue queue = new LinkedList(); queue.offer(root); // 首先根结点先入队 while (!queue.isEmpty()) {     int size = queue.size(); // 第一层的size = 1;     List ans = new ArrayList();     while (size != 0) {  TreeNode tmp = queue.poll();  ans.add(tmp.val); // 把当前从栈中出来的元素放到我们定义的List中,因为是同一层的,在一个循环里的  if (tmp.left != null) {      queue.offer(tmp.left);  }  if (tmp.right != null) {      queue.offer(tmp.right);  }  --size;     }     ret.add(ans); } return ret;    }

三、其他一些常见操作

🍑获取树中结点的个数

📝遍历 

 // 获取树的结点的个数,遍历    public static int nodeSize;    void size2(TreeNode root) { if (root == null) return; ++nodeSize; size2(root.left); size2(root.right);    }

递归的大致流程如图所示: 

   

📝递归,转换为左右子问题求解

// 获取树中节点的个数,子问题求解    int size(TreeNode root) { if (root == null) return 0; int leftSize = size(root.left);  // 递归到root.left == null 开始返回:return 0; int rightSize = size(root.right); // 当leftSize和rightSize都第一次递归结束(即道理叶结点出,此时叶子结点的左右子节点都为空之间返回0,但你能说结点为0吗 // 肯定不能虽然当前结点的左右子结点数目都为0,但别忘了当前结点也是算一个结点呀!!!所以要+1 return leftSize + rightSize + 1;    }


🍑获取叶子结点的个数 

// 遍历思路—获取叶子节点的个数    public static int leafSize;    // 因为getLeafNodeCount1函数中有递归,所以我们的nodeSize不能定义到函数里    // (因为递归不断调用我们的函数,相当于会给nodeSize不断的赋初始值,所以我们对nodeSize累加的值会被冲掉    int getLeafNodeCount1(TreeNode root){ if (root == null) return 0; if (root.left == null && root.right == null) {     ++leafSize; } getLeafNodeCount1(root.left); // 其实遍历也离不开递归 getLeafNodeCount1(root.right); return leafSize;    }

其实这个和上面的递归思路也是一样的

// 子问题思路-求叶子结点个数    int getLeafNodeCount2(TreeNode root) { if (root == null) return 0; if (root.left == null && root.right == null) return 1; return getLeafNodeCount2(root.left) + getLeafNodeCount2(root.right); // 返回左子树的叶子结点的个数 + 右子树叶子结点的个数    }


🍑获取第k层结点的个数 

// 获取第K层节点的个数    int getKLevelNodeCount(TreeNode root, int k) { if (root == null) return 0; if (k == 1) return 1; return getKLevelNodeCount(root.left, k - 1) + getKLevelNodeCount(root.right, k - 1);    }


🍑获取二叉树的高度 

递归思路,二叉树天生就适合递归,因为二叉树的定义就用到了递归的概念 

   

 // 获取二叉树的高度——时间复杂度O(N)    int getHeight(TreeNode root) { if (root == null) return 0; // 下面这样写虽然对,但应为多递归了一次;在oj上可能超时 //return (getHeight(root.left) > getHeight(root.right)) ? getHeight(root.left) + 1 : getHeight(root.right) + 1; // 所以这样写,用变量把我们递归的结果保存下来,这样在三目运算符?后就不用在重复的调用递归求高度了 int leftHeight = getHeight(root.left); int rightHeight = getHeight(root.right); return (leftHeight > rightHeight) ? leftHeight + 1 : rightHeight + 1; // 做左右子树的最大深度加1    }


🍑检测value结点是否存在 

// 检测值为value的元素是否存在——转换为子问题    TreeNode find(TreeNode root, char val) { if(root == null) return null; if(root.val == val) return root; TreeNode treeNode1 = find(root.left, val); TreeNode treeNode2 = find(root.right, val); if (treeNode1 == null) return treeNode2; else return treeNode1;    }


 四、完整的代码

import java.util.ArrayList;import java.util.LinkedList;import java.util.List;import java.util.Queue;// 树类public class MyBinaryTree {    // 结点类    class TreeNode { public char val;  // 储存该结点的数值 public TreeNode left; //左孩子的引用,即该结点的左子结点 public TreeNode right; //右孩子的引用 // 三种对树的结点的构造方法(每一种方法所传参数都是不同的) TreeNode() { } // 其实我们接下来用到的都是这一种构造方法,因为接下来我们只是在构造方法中给该结点赋值,至于该结点的左右子结点,我们是手动给他们建立联系的 TreeNode(char val) {     this.val = val; } TreeNode(char val, TreeNode left, TreeNode right) {     this.val = val;     this.left = left;     this.right = right; } @Override  // 重写toString方法,一般接下来能够之间打印树的结点 public String toString() {     return "TreeNode{" +      "val=" + val +      '}'; }    }    // 其实这不是二叉树真正的构建方式,但我们的目的是快速构建一个二叉树,学习的二叉树的一些方法(这是我们的侧重点)    // 至于真正的构建方法,我们之后的博客会讲到,本篇博客我们重点了解二叉树所对应的各种独特的方法    public TreeNode createBinaryTree(){ TreeNode node1 = new TreeNode('A'); TreeNode node2 = new TreeNode('B'); TreeNode node3 = new TreeNode('C'); TreeNode node4 = new TreeNode('D'); TreeNode node5 = new TreeNode('E'); TreeNode node6 = new TreeNode('F'); TreeNode root = node1;  // 我们采用了穷举法来快速的构建一棵二叉树 node1.left = node2; node1.right = node3; node2.left = node4; node2.right = node5; node3.left = node6; return root;     // 返回我们构建树的根结点    }    // 前序遍历——》根->左子树->右子树    public void preOrder(TreeNode root) { if (root == null) return; System.out.print(root.val + " "); postOrder(root.left); postOrder(root.right);    }    // 中序遍历——》左子树—>根->右子树    public void inOrder(TreeNode root) { if (root == null) return; inOrder(root.left); System.out.print(root.val + " "); inOrder(root.right);    }    // 后续遍历——》左子树->右子树—>根    public void postOrder(TreeNode root) { if (root == null) return; postOrder(root.left); postOrder(root.right); System.out.print(root.val + " ");    }    // 获取树的结点的个数,遍历    public static int nodeSize;    void size2(TreeNode root) { if (root == null) return; ++nodeSize; size2(root.left); size2(root.right);    }    // 获取树中节点的个数,子问题求解    int size(TreeNode root) { if (root == null) return 0; int leftSize = size(root.left);  // 递归到root.left == null 开始返回:return 0; int rightSize = size(root.right); // 当leftSize和rightSize都第一次递归结束(即道理叶结点出,此时叶子结点的左右子节点都为空之间返回0,但你能说结点为0吗 // 肯定不能虽然当前结点的左右子结点数目都为0,但别忘了当前结点也是算一个结点呀!!!所以要+1 return leftSize + rightSize + 1;    }    // 遍历思路—获取叶子节点的个数    public static int leafSize;    // 因为getLeafNodeCount1函数中有递归,所以我们的nodeSize不能定义到函数里    // (因为递归不断调用我们的函数,相当于会给nodeSize不断的赋初始值,所以我们对nodeSize累加的值会被冲掉    int getLeafNodeCount1(TreeNode root){ if (root == null) return 0; if (root.left == null && root.right == null) {     ++leafSize; } getLeafNodeCount1(root.left); // 其实遍历也离不开递归 getLeafNodeCount1(root.right); return leafSize;    }    // 子问题思路-求叶子结点个数    int getLeafNodeCount2(TreeNode root) { if (root == null) return 0; if (root.left == null && root.right == null) return 1; return getLeafNodeCount2(root.left) + getLeafNodeCount2(root.right); // 返回左子树的叶子结点的个数 + 右子树叶子结点的个数    }    // 获取第K层节点的个数    int getKLevelNodeCount(TreeNode root, int k) { if (root == null) return 0; if (k == 1) return 1; return getKLevelNodeCount(root.left, k - 1) + getKLevelNodeCount(root.right, k - 1);    }    // 获取二叉树的高度——时间复杂度O(N)    int getHeight(TreeNode root) { if (root == null) return 0; // 下面这样写虽然对,但应为多递归了一次;在oj上可能超时 //return (getHeight(root.left) > getHeight(root.right)) ? getHeight(root.left) + 1 : getHeight(root.right) + 1; // 所以这样写,用变量把我们递归的结果保存下来,这样在三目运算符?后就不用在重复的调用递归求高度了 int leftHeight = getHeight(root.left); int rightHeight = getHeight(root.right); return (leftHeight > rightHeight) ? leftHeight + 1 : rightHeight + 1; // 做左右子树的最大深度加1    }    // 检测值为value的元素是否存在——转换为子问题    TreeNode find(TreeNode root, char val) { if(root == null) return null; if(root.val == val) return root; TreeNode treeNode1 = find(root.left, val); TreeNode treeNode2 = find(root.right, val); if (treeNode1 == null) return treeNode2; else return treeNode1;    }    //层序遍历    public void levelOrder(TreeNode root) { if(root == null) return; Queue queue = new LinkedList(); // 我们需要借助一个队列, queue.offer(root); // 先把根结点放到队列中,根结点相当于是第一层 while (!queue.isEmpty()) {     // 借助tmp来把当前结点的左右子树给放到队列中     // 注意随着队列元素的不断弹出,tmp是在动态变化的,所以才能变量完——整棵二叉树     TreeNode tmp = queue.poll();  // 因为这一层的元素入队列的时候,是通过上一层元素的对左右子树操作来实现的,入队列是一层一层入的     System.out.print(tmp.val + " ");     if (tmp.left != null) {  queue.offer(tmp.left);  // 把当前从队列中弹出的结点的左子结点放到队列中     }     if (tmp.right != null) {  queue.offer(tmp.right); // 右子结点     } }    }    // 力扣oj题目——层序遍历,因为要求不同所以我们的具体做法有些不一样,但基本的思想相同    public List<List> levelOrder1(TreeNode root) { List<List> ret = new ArrayList(); if (root == null) return ret; // Queue queue = new LinkedList(); queue.offer(root); // 首先根结点先入队 while (!queue.isEmpty()) {     int size = queue.size(); // 第一层的size = 1;     List ans = new ArrayList();     while (size != 0) {  TreeNode tmp = queue.poll();  ans.add(tmp.val); // 把当前从栈中出来的元素放到我们定义的List中,因为是同一层的,在一个循环里的  if (tmp.left != null) {      queue.offer(tmp.left);  }  if (tmp.right != null) {      queue.offer(tmp.right);  }  --size;     }     ret.add(ans); } return ret;    }}

 五、测试代码

public class TestMyBinaryTree {    public static void main(String[] args) { MyBinaryTree myBinaryTree = new MyBinaryTree(); MyBinaryTree.TreeNode root = myBinaryTree.createBinaryTree(); myBinaryTree.size2(root); System.out.println(MyBinaryTree.nodeSize); System.out.print("前序遍历的结果是:"); myBinaryTree.preOrder(root); // 前序遍历 System.out.println(); System.out.print("中序遍历的结果是:"); myBinaryTree.inOrder(root);   // 中序遍历 System.out.println(); System.out.print("后序遍历的结果是:"); myBinaryTree.postOrder(root); // 后序遍历 System.out.println(); System.out.println("=========我是分割线========="); System.out.println("在树中,对B的查找结果是:" + myBinaryTree.find(root, 'B')); System.out.println("在树中,对d的查找结果是:" + myBinaryTree.find(root, 'd')); System.out.println("当前树中的最大深度为:" + myBinaryTree.getHeight(root)); System.out.println("=========我是分割线========="); System.out.println("当前树中的结点个数是:" + myBinaryTree.size(root)); System.out.println("当前树中的叶子结点个数:" + myBinaryTree.getLeafNodeCount1(root)); System.out.println("当前树中第2层的结点个数为:" + myBinaryTree.getKLevelNodeCount(root, 2)); System.out.print("层序遍历的结果为:"); myBinaryTree.levelOrder(root);    }}

测试结果