Diameter of Binary Tree in scala

Sumit Rawal answered on September 2, 2023 Popularity 1/10 Helpfulness 1/10

answer Diameter of Binary Tree in scala

Diameter of Binary Tree in scala

Comment

object DiameterOfBinaryTree {
  case class TreeNode(value: Int, left: Option[TreeNode], right: Option[TreeNode])

  def diameterOfBinaryTree(root: Option[TreeNode]): Int = {
    var diameter = 0

    def depth(node: Option[TreeNode]): Int = {
      node match {
        case Some(n) =>
          val leftDepth = depth(n.left)
          val rightDepth = depth(n.right)
          diameter = math.max(diameter, leftDepth + rightDepth)
          math.max(leftDepth, rightDepth) + 1
        case None => 0
      }
    }

    depth(root)
    diameter
  }

  def main(args: Array[String]): Unit = {
    // Example binary tree:
    //        1
    //       / \
    //      2   3
    //     / \
    //    4   5
    val tree = Some(TreeNode(1, Some(TreeNode(2, Some(TreeNode(4, None, None)), Some(TreeNode(5, None, None)))), Some(TreeNode(3, None, None))))
    val result = diameterOfBinaryTree(tree)
    println(s"Diameter of the binary tree: $result")
  }
}

Popularity 1/10 Helpfulness 1/10 Language scala

Source: chat.openai.com

Tags: binary-tree scala

Link to this answer
Share Copy Link

Contributed on Sep 02 2023

Sumit Rawal

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

Diameter of Binary Tree in scala

Contents

More Related Answers

Diameter of Binary Tree in scala

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.