class Solution {
    func arrangeCoins(_ n: Int) -> Int {
        let sq = Int(sqrt(Double(n * 2)))
        let diff = sq * (sq + 1) / 2
        if n >= diff {
            return sq
        } else {
            return sq - 1
        }
    }
}