Create 0538-convert-bst-to-greater-tree.kt

RatanKokal · Oct 28, 2023 · 66292aa · 66292aa
1 parent 655c2b2
commit 66292aa
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/kotlin/0538-convert-bst-to-greater-tree.kt b/kotlin/0538-convert-bst-to-greater-tree.kt
@@ -0,0 +1,19 @@
+class Solution {
+    fun convertBST(root: TreeNode?): TreeNode? {
+        var curSum = 0
+
+        fun dfs(node: TreeNode?) {
+            node?: return
+
+            dfs(node.right)
+            val temp = node.`val`
+            node.`val` += curSum
+
+            curSum += temp
+            dfs(node.left)
+        }
+
+        dfs(root)
+        return root
+    }
+}