From 1080f284001174bc8fdb4f565fd7ba2ef86fcabe Mon Sep 17 00:00:00 2001
From: Matej Focko <me@mfocko.xyz>
Date: Mon, 6 May 2024 20:51:17 +0200
Subject: [PATCH] =?UTF-8?q?kt:=20add=20=C2=AB2487.=20Remove=20Nodes=20From?=
 =?UTF-8?q?=20Linked=20List=C2=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Matej Focko <me@mfocko.xyz>
---
 kt/remove-nodes-from-linked-list.kt | 71 +++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 kt/remove-nodes-from-linked-list.kt

diff --git a/kt/remove-nodes-from-linked-list.kt b/kt/remove-nodes-from-linked-list.kt
new file mode 100644
index 0000000..41d5ded
--- /dev/null
+++ b/kt/remove-nodes-from-linked-list.kt
@@ -0,0 +1,71 @@
+class ListNode(var `val`: Int) {
+    var next: ListNode? = null
+}
+
+fun toListNode(vararg xs: Int): ListNode? {
+    if (xs.isEmpty()) {
+        return null
+    }
+
+    val dummy = ListNode(0)
+    var node = dummy
+
+    for (x in xs) {
+        val next = ListNode(x)
+        node.next = next
+        node = next
+    }
+
+    return dummy.next
+}
+
+fun linkedListEquals(head: ListNode?, xs: List<Int>): Boolean {
+    var node = head
+    for ((i, x) in xs.withIndex()) {
+        if (node == null) {
+            println("[DEBUG] $x is expected at index $i in the linked list, but is not present")
+            return false
+        } else if (node.`val` != x) {
+            println("[DEBUG] $x is expected at index $i in the linked list, but ${node.`val`} is present")
+            return false
+        }
+
+        node = node.next
+    }
+
+    if (node != null) {
+        println("[DEBUG] Linked list is longer than expected")
+    }
+
+    return node == null
+}
+
+class Solution {
+    fun removeNodes(head: ListNode?): ListNode? = removeNodesRec(head).first
+
+    private fun removeNodesRec(head: ListNode?): Pair<ListNode?, Int> {
+        if (head == null) {
+            return head to Int.MIN_VALUE
+        } else if (head.next == null) {
+            return head to head.`val`
+        }
+
+        val (next, rightMax) = removeNodesRec(head.next)
+
+        var node = head
+        if (head.`val` < rightMax) {
+            node = next
+        } else {
+            node.next = next
+        }
+
+        return node to maxOf(head.`val`, rightMax)
+    }
+}
+
+fun main() {
+    val s = Solution()
+
+    check(linkedListEquals(s.removeNodes(toListNode(5, 2, 13, 3, 8)), listOf(13, 8)))
+    check(linkedListEquals(s.removeNodes(toListNode(1, 1, 1, 1)), listOf(1, 1, 1, 1)))
+}