leetcode_441 排列硬币

要求

你总共有 n 枚硬币，并计划将它们按阶梯状排列。对于一个由 k 行组成的阶梯，其第 i 行必须正好有 i 枚硬币。阶梯的最后一行 可能 是不完整的。

给你一个数字 n ，计算并返回可形成 完整阶梯行 的总行数。

示例 1：

输入：n = 5
输出：2
解释：因为第三行不完整，所以返回 2 。

示例 2：

输入：n = 8
输出：3
解释：因为第四行不完整，所以返回 3 。

核心代码

class Solution:
    def arrangeCoins(self, n: int) -> int:
        lo,hi = 0,n
        while lo <= hi:
            mid = (lo + hi) // 2
            total = (mid + 1) * mid // 2
            if total > n:
                hi = mid - 1
            elif total < n:
                lo  = mid + 1 
            else:
                return mid
        return lo - 1

解题思路：这个题实际上是在找k * (k + 1) /2 <= n,我们就是想要找这个k值，所以是一个搜索问题，我们可以使用二分查找的方式进行查找。