Параметр функции Kotlin: Val не может быть переназначен
Я написал Красно-черное дерево в Котлине. Fun insertFixup восстанавливает баланс после вставки нового элемента (z: Node? Is new element). Алгоритм балансировки деревьев взят из здесь (стр. 2-3). Проблема заключается в том, что Kotlin не позволяет мне переназначать z на z.parent и z.parent.parent. Я хочу, чтобы z был указателем. Вопрос в том, как заставить Котлина понять, чего я от него хочу?
class Node(key: Int) {...}
class BinarySearchTree {
var root: Node? = null
fun insert(newNode: Node) {...}
fun RotateLeft(x: Node?) {...}
fun RotateRight(x: Node?) {...}
fun insertFixup(z: Node?) {
var y: Node?
while (z?.parent?.color == "RED") {
if (z?.parent == z?.parent?.parent?.left) {
y = z?.parent?.parent?.right
if (y?.color == "RED") {
z?.parent?.color = "BLACK"
y?.color = "BLACK"
z?.parent?.parent?.color = "RED"
z = z?.parent?.parent
}
if (z == z?.parent?.right) {
z = z?.parent
RotateLeft(z)
z?.parent?.color = "BLACK"
z?.parent?.parent?.color = "RED"
RotateRight(z?.parent?.parent)
}
} else {
y = z?.parent?.parent?.left
if (y?.color == "RED") {
z?.parent?.color = "BLACK"
y?.color = "BLACK"
z?.parent?.parent?.color = "RED"
z = z?.parent?.parent
}
if (z != z?.parent?.left) {
z = z?.parent
RotateLeft(z)
z?.parent?.color = "BLACK"
z?.parent?.parent?.color = "RED"
RotateRight(z?.parent?.parent)
}
}
}
root?.color = "BLACK"
}
}
fun main(args: Array<String>) {
val bst = BinarySearchTree()
while (true) {
var newNode = Node(readLine()!!.toInt())
bst.insert(newNode)
bst.insertFixup(newNode)
}
}
UPD: Спасибо всем! Все ответы были полезны, и я нашел решение в ваших ответах.
Ответы
Ответ 1
Параметры функции в Котлин для чтения только val
внутри функции, поэтому z
здесь всегда будет ссылаться на исходный объект, который был передан в.
Если вам нужно изменить то, на что указывает, пока ваша функция запущена, вам нужно будет сделать локальную копию в начале функции, а затем вы можете сделать это var
.
Например, вы можете запустить свою функцию следующим образом:
fun insertFixup(_z: Node?) {
var z = _z
Ответ 2
Параметры функции Kotlin доступны только для чтения и не могут быть назначены.
Однако вы можете создать объект ReadWriteProperty
для передачи в insertFixup
для получения/настройки newNode
:
...
class BinarySearchTree {
...
fun insertFixup(zProperty: ReadWriteProperty<Any?, Node?>) {
var z by zProperty
...
fun main(args: Array<String>) {
val bst = BinarySearchTree()
var newNode: Node? = null
val newNodeProperty = object : ReadWriteProperty<Any?, Node?> {
override operator fun getValue(thisRef: Any?, property: KProperty<*>): Node? {
return newNode
}
override operator fun setValue(thisRef: Any?, property: KProperty<*>,
value: Node?) {
newNode = value
}
}
while (true) {
newNode = Node(readLine()!!.toInt())
bst.insert(newNode!!)
bst.insertFixup(newNodeProperty)
}
}
И если вы хотите использовать свойство вместо переменной, вы можете использовать ссылку на свойство для получения/установки newNode
из insertFixup
:
...
class BinarySearchTree {
...
fun insertFixup(zProperty: KMutableProperty0<Node?>) {
var z by zProperty
...
var newNode: Node? = null
fun main(args: Array<String>) {
val bst = BinarySearchTree()
while (true) {
newNode = Node(readLine()!!.toInt())
bst.insert(newNode!!)
bst.insertFixup(::newNode)
}
}
// the following allow 'KMutableProperty0' to be used as a read/write delegate
operator fun <T> KProperty0<T>.getValue(thisRef: Any?, property: KProperty<*>): T = get()
operator fun <T> KMutableProperty0<T>.setValue(thisRef: Any?, property: KProperty<*>,
value: T) = set(value)
Ответ 3
Я столкнулся с этой проблемой также. Я создал класс данных и передал класс данных в качестве параметра, который затем мог бы использовать для изменения его свойств.
data class SomeDataClass(
val x: Int,
val y: Int,
val z: Int
)
fun someMethod(someDataClass: SomeDataClass) {
someDataClass.z = 23 //whatever Int value you please
// more computations...
someDataClass.z = 67 // or whatever new value you need to assign.
}
fun parentMethod() {
val someDataClass = SomeDataClass()
someMethod(someDataClass)
val newZValue = someDataClass.z // someDataClass holds modified data from above
// method
}