【二叉树】如何构建一个包含大量随机数节点的二叉树测试用例

  • 前言
  • 一、案例准备
  • 二、自动生成随机二叉树工具类（TreegenerateUtils）
  • 三、如何调用随机二叉树工具类（TreegenerateUtils）？

前言

今天笔者在测试有关二叉树的测试用例时，发现一点一点给节点添加孩子操作十分繁琐，于是写了一个自动生成二叉树测试用例函数，供大家参考。

一、案例准备

准备了一个二叉树节点类Node如下图所示

二、自动生成随机二叉树工具类（TreegenerateUtils）

  如下面代码所示，定义了一个TreegenerateUtil工具类，其中包含有静态方法generate自动生成随机二叉树，其中该方法有两个重载实现，该方法返回一个随机二叉树实例的根节点，其中两个重载方法都需要四个参数，其中前三个参数相同为currentDepth（树的深度），MaxDepth(树的最大深度），MaxValue(树的每个节点可以达到的最大值)，最后一个参数第一个重载方法为Random类实例，目的是使得每次生成的随机树不同，而第二个重载方法的参数为Int数字,跟第一个重载方法也是同样的目的

package net.mooctest;import java.util.Random;/** * @ClassName TreeUtils * @Description * @Author chougou * @Date 2023年11月10日 21:19 * @Version 1.0 */public class TreegenerateUtils {public static Node generate(int currentDepth, int MaxDepth, int MaxValue, Random random1) { //currentDepth为当前深度,MaxDepth为树的最大深度,MaxValue为节点可以达到的最大数,均可自行调节,random为随机数种子//因为每次递归深度不同，因此currentDepth+random可作为随机数种子if (currentDepth >= MaxDepth) { //当递归树的深度超过MaxDepth时候,递归结束return new Node(random1.nextInt(MaxValue));} else {Node node = new Node(random1.nextInt(MaxValue));//随机添加左孩子或者右孩子或者左右孩子或者无孩子//1，只添加左孩子int temp = random1.nextInt(100);if (temp % 3 == 0) {Node nodeleft = generate(currentDepth + 1, MaxDepth, MaxValue, random1);node.left = nodeleft;}//2.只添加右孩子if (temp % 3 == 1) {Node noderight = generate(currentDepth + 1, MaxDepth, MaxValue, random1);node.right = noderight;}//3.左右孩子均添加if (temp % 3 == 2) {Node noderight = generate(currentDepth + 1, MaxDepth, MaxValue, random1);Node nodeleft = generate(currentDepth + 1, MaxDepth, MaxValue, random1);node.left = nodeleft;node.right = noderight;}//4.最后一种即什么也不添加return node;}}public static Node generate(int currentDepth, int MaxDepth, int MaxValue, int random) { //currentDepth为当前深度,MaxDepth为树的最大深度,MaxValue为节点可以达到的最大数,均可自行调节,random为随机数种子Random random1 = new Random(currentDepth + random);//因为每次递归深度不同，因此currentDepth+random可作为随机数种子if (currentDepth >= MaxDepth) { //当递归树的深度超过MaxDepth时候,递归结束return new Node(random1.nextInt(MaxValue));} else {Node node = new Node(random1.nextInt(MaxValue));//随机添加左孩子或者右孩子或者左右孩子或者无孩子//1，只添加左孩子int temp = random1.nextInt(100);if (temp % 3 == 0) {Node nodeleft = generate(currentDepth + 1, MaxDepth, MaxValue, random);node.left = nodeleft;}//2.只添加右孩子if (temp % 3 == 1) {Node noderight = generate(currentDepth + 1, MaxDepth, MaxValue, random);node.right = noderight;}//3.左右孩子均添加if (temp % 3 == 2) {Node noderight = generate(currentDepth + 1, MaxDepth, MaxValue, random);Node nodeleft = generate(currentDepth + 1, MaxDepth, MaxValue, random);node.left = nodeleft;node.right = noderight;}//4.最后一种即什么也不添加return node;}}}

三、如何调用随机二叉树工具类（TreegenerateUtils）？

  如下面程序示例，通过调用TreegenerateUtils的静态函数generate生成4个根节点，其中每个根节点均不相同

 @Testpublic void testTreegenerateUtils(){//下面生成四个最大深度为10,节点可达到最大值为100的四个随机二叉树实例Random random1=new Random(); Node root1=TreegenerateUtils.generate(1,10,100,random1.nextInt()); Node root2=TreegenerateUtils.generate(1,10,100,random1.nextInt()); Node root3=TreegenerateUtils.generate(1,10,100,random1); Node root4=TreegenerateUtils.generate(1,10,100,random1);}

运行结果展示：（每个根节点均不相同）