Form Triangle - KOTLIN

// Author : Saurav Kalsoor
// Form Triangle - KOTLIN

import java.util.*

var sc: Scanner = Scanner(System.`in`)

fun main() {
    val n: Long = sc.nextLong()
    val result = formTriangle(n)
    println(result)
}

fun formTriangle(n: Long): Long {
    var lo: Long = 1
    var hi = n
    var result: Long = 1
    while (lo <= hi) {
        val mid = lo + (hi - lo) / 2
        val requiredChocolates = mid * (mid + 1) / 2
        if (requiredChocolates <= n) {
            result = mid
            lo = mid + 1
        } else {
            hi = mid - 1
        }
    }
    return result
}