Fruits into Basket (https://leetcode.com/problems/fruit-into-baskets/)

fruits.kt

fun totalFruit(tree: IntArray): Int {
    val collectedFruitSet = ArrayList<Int>()
    var lastMax = 0

    for (loopingIndex in 0 until tree.size) {
        if (tree.distinct().count() <= 2) {
            lastMax = tree.size
            break
        } else if (collectedFruitSet.distinct().count() < 2 || (
                    collectedFruitSet.distinct().count() == 2 && collectedFruitSet.contains(tree[loopingIndex])
                    )
        ) {
            collectedFruitSet.add(tree[loopingIndex])
        } else {
            var tempIndex = 0

            while (collectedFruitSet.distinct().count() > 1) {
                if (!collectedFruitSet.contains(tree[loopingIndex])) collectedFruitSet.removeAt(0)
                else break

                tempIndex++
            }
            collectedFruitSet.add(tree[loopingIndex])
        }

        lastMax = max(lastMax, collectedFruitSet.size)
    }

    return lastMax
}