Scala LeetCode 965. 单值二叉树

最新推荐文章于 2024-04-16 23:23:44 发布

原创最新推荐文章于 2024-04-16 23:23:44 发布 · 370 阅读

CC 4.0 BY-SA版权

本文介绍了一个使用Scala语言实现的算法，用于检查一棵二叉树是否为单值二叉树，即树中所有节点的值是否相同。通过递归遍历树的每个节点并与根节点的值进行比较，若所有节点值一致则返回真，否则返回假。文章还讨论了Scala中类的初始化、集合类的使用及函数特性。

https://leetcode-cn.com/problems/univalued-binary-tree/

class TreeNode(var _value: Int) {
  var value: Int = _value
  var left: TreeNode = _
  var right: TreeNode = _
}
object Main {
  def isUnivalTree(root: TreeNode): Boolean = {
    var b = List[TreeNode]()
    val x = root._value
    b = root :: b
    while (b.nonEmpty) {
      val t = b.head
      b = b.tail
      if (t.left != null) {
        if (t.left._value != x) {
          return false
        }
        b = t.left :: b
      }
      if (t.right != null) {
        if (t.right._value != x) {
          return false
        }
        b = t.right :: b
      }
    }
    true
  }
  def main(args: Array[String]): Unit = {
    val root = new TreeNode(0)
    root.right = new TreeNode(0)
    assert(isUnivalTree(root), true)
  }
}

继续总结