Create 0129-sum-root-to-leaf-numbers.kt

neetcode-gh · Mar 14, 2023 · 2f32496 · 2f32496
1 parent 6215af4
commit 2f32496
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/kotlin/0129-sum-root-to-leaf-numbers.kt b/kotlin/0129-sum-root-to-leaf-numbers.kt
@@ -0,0 +1,25 @@
+class Solution {
+    fun sumNumbers(root: TreeNode?): Int {
+        var res = 0
+
+        fun dfs(root: TreeNode?, current: Int) {
+            root?: return
+
+            var new = current * 10
+            new += root.value
+
+            if(root.left == null && root.right == null) {
+                res += new
+                return
+            }
+
+            dfs(root.left, new)
+            dfs(root.right, new)
+        }
+
+        dfs(root, 0)
+        return res
+    }
+
+    val TreeNode.value get()= this.`val`
+}