Skip to content

Instantly share code, notes, and snippets.

@kafri8889
Created July 27, 2023 11:04
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save kafri8889/82d0ca18b48750ea7db400cd353ca70e to your computer and use it in GitHub Desktop.
Save kafri8889/82d0ca18b48750ea7db400cd353ca70e to your computer and use it in GitHub Desktop.
Class for undo/redo feature (linked list)
class Unre<T> {
class UnreListNode<T>(var value: T) {
var next: UnreListNode<T>? = null
var prev: UnreListNode<T>? = null
override fun toString(): String {
return "ListNode(${value}, ${next?.toString()})"
}
}
private var head: UnreListNode<T>? = null
private var tail: UnreListNode<T>? = null
fun isEmpty(): Boolean = head == null
fun canUndo(): Boolean = tail?.prev != null
fun canRedo(): Boolean = tail?.next != null
fun add(element: T) {
UnreListNode(element).let {
if (isEmpty()) head = it.also { tail = it }
else it.apply { prev = tail?.apply { next = it.also { tail = it } } }
}
}
fun undo(): T {
return if (canUndo()) {
tail = tail!!.prev
tail!!.value
} else tail!!.value
}
fun redo(): T {
return if (canRedo()) {
tail = tail!!.next
tail!!.value
} else tail!!.value
}
}
// ------------------------------ Example -----------------------------------------------------------
fun main() {
val unre = Unre<String>()
unre.add("Ena")
unre.add("Enami")
unre.add("Enami Asa")
unre.add("Enami Asa <3")
println(unre.undo()) // "Enami Asa"
println(unre.undo()) // "Enami"
println(unre.redo()) // "Enami Asa"
println(unre.redo()) // "Enami Asa <3"
unre.undo()
unre.undo()
println(unre.undo()) // "Ena"
println(unre.redo()) // "Enami"
unre.add("Enami Lisa")
println(unre.undo()) // Enami
println(unre.redo()) // Enami Lisa
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment