【scala】含有list子集的递归方法记录层数深度

样例使用scala语言编写。
特点：
RecursiveData对象中包罗一个成员child含有多个RecursiveData子对象。
思绪：
递归方法中传入一个：int作为parent的深度，传入一个set记录以是深度聚集。
同一深度int+1不会影响雷同深度的效果，set又可以将同一深度层去重。
容易犯错误：使用单独对象记录，应为是递归是树形布局，随着层数和枝叶增长，会多盘算，如recWrong方法就是错误范例。

1. import org.scalatest.funsuite.AnyFunSuite
3. import java.util.concurrent.atomic.AtomicInteger
4. import scala.collection.mutable
6. case class RecursiveData(name: String, child: List[RecursiveData])
8. class RecursiveTest extends AnyFunSuite {
10.   val r41 = RecursiveData("41", List.empty)
11.   val r42 = RecursiveData("42", List.empty)
12.   val r43 = RecursiveData("43", List.empty)
13.   val r44 = RecursiveData("44", List.empty)
14.   val r31 = RecursiveData("31", List(r41, r42))
15.   val r32 = RecursiveData("32", List(r43, r44))
16.   val r33 = RecursiveData("33", List.empty)
17.   val r34 = RecursiveData("34", List.empty)
18.   val r35 = RecursiveData("35", List.empty)
19.   val r36 = RecursiveData("36", List.empty)
20.   val r37 = RecursiveData("37", List.empty)
21.   val r21 = RecursiveData("21", List(r31, r32, r33))
22.   val r22 = RecursiveData("22", List(r34, r35))
23.   val r23 = RecursiveData("23", List(r36, r37))
24.   val r11 = RecursiveData("11", List(r21, r22, r23))
26.   def rec(data: RecursiveData, parentDepth: Int, depths: mutable.Set[Integer]): Unit = {
27.     val currDepth = parentDepth + 1
28.     depths.add(currDepth)
29.     data.child.foreach(c => {
30.       rec(c, currDepth, depths)
31.     })
32.   }
34.   test("test-recursive-right") {
35.     val depths: mutable.Set[Integer] = mutable.Set.empty
36.     rec(r11, 0, depths)
37.     println(s"current depths:${depths},and max depths:${depths.max}")
38.     // current depths:Set(1, 2, 3, 4),and max depths:4
39.   }
41.   def recWrong(data: RecursiveData, depth: AtomicInteger): Unit = {
42.     data.child.foreach(c => {
43.       recWrong(c, depth)
44.     })
45.     if (data.child.nonEmpty) {
46.       // 错误思路: 每次增加减去相应的子成员增加量
47.       depth.set(depth.get() - (data.child.size - 1))
48.     }
49.     depth.getAndIncrement()
50.   }
52.   test("test-recursive-wrong") {
53.     val depth: AtomicInteger = new AtomicInteger
54.     recWrong(r11, depth)
55.     println(depth)
56.     // 7
57.   }
58. }

复制代码