CowsGameMain.kt

import java.util.TreeSet

enum class LetterStatus {
    NOT_CORRECT,
    CONTAINS,
    CORRECT
}
// a -> ответ
// b -> строка, которая с ним сравнивается
fun compareWords(a: String, b: String): List<LetterStatus> {
    // тут нужно вспомнить про бинарные AVL деревья
    val treeSet = TreeSet<Char>()

    // доступ можно ускорить, если придумать как гарантировать, добавление только уникальных букв
    treeSet.addAll(a.toSet())

    val ans = mutableListOf<LetterStatus>()
    for (i in a.indices){
        ans.add(
            if(a[i] == b[i]){
                LetterStatus.CORRECT
            } else if (treeSet.contains(b[i])){
                LetterStatus.CONTAINS
            } else {
                LetterStatus.NOT_CORRECT
            }
        )
    }

    return ans
}

fun main() {
    println(
        compareWords("HELLO", "HLLAO")
    )
}