437/kt-nopass

Author: zheeeng

437.path-sum-iii.0.[nopass].kt

@@ -0,0 +1,40 @@
+/*
+ * @lc app=leetcode id=437 lang=kotlin
+ *
+ * [437] Path Sum III
+ */
+/**
+ * Example:
+ * var ti = TreeNode(5)
+ * var v = ti.`val`
+ * Definition for a binary tree node.
+ * class TreeNode(var `val`: Int) {
+ *     var left: TreeNode? = null
+ *     var right: TreeNode? = null
+ * }
+ */
+// TLE
+class Solution_pathSum_0 {
+    fun getPaths(root: TreeNode?): Pair<List<List<Int>>, List<List<Int>>> {
+        if (root == null) return Pair(listOf<List<Int>>(), listOf<List<Int>>())
+
+        val self = listOf(listOf(root.`val`))
+
+        val (leftMut, leftInmmut) = getPaths(root.left)
+
+        val (rightMut, rightInmmut) = getPaths(root.right)
+
+        val nextMut = (leftMut + rightMut)
+            .filter { it.size > 0 }
+            .map {
+                listOf(root.`val`) + it
+            } + self
+
+        val nextImmut = nextMut + leftInmmut + rightInmmut
+        return Pair(nextMut, nextImmut)
+    }
+
+    fun pathSum(root: TreeNode?, sum: Int): Int {
+        return getPaths(root).second.filter { it.sum() == sum }.size
+    }
+}