Commit fdfa05e

committed

scala array argmax

1 parent 7efa748 commit fdfa05eCopy full SHA for fdfa05e

File tree

+11

-0

lines changed

+11

-0

lines changed

Lines changed: 11 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,11 @@`
	`1`	`+object Solution {`
	`2`	`+ def constructMaximumBinaryTree(nums: Array[Int]): TreeNode = {`
	`3`	`+ val maxIdx = nums.zipWithIndex.maxBy(_._1)._2`
	`4`	`+ val root = TreeNode(nums(maxIdx))`
	`5`	`+ if (maxIdx == 0) root.left = null`
	`6`	`+ else root.left = constructMaximumBinaryTree(nums.slice(0, maxIdx))`
	`7`	`+ if (maxIdx == nums.length - 1) root.right = null`
	`8`	`+ else root.right = constructMaximumBinaryTree(nums.slice(maxIdx + 1, nums.length))`
	`9`	`+ root`
	`10`	`+ }`
	`11`	`+}`

Comments

(0)