441 排列硬币（模拟、解方程、二分）_44411把1222次dabad33-CSDN博客

本文链接：https://blog.csdn.net/qq_39445165/article/details/118370906

1. 问题描述：

你总共有 n 枚硬币，你需要将它们摆成一个阶梯形状，第 k 行就必须正好有 k 枚硬币。给定一个数字 n，找出可形成完整阶梯行的总行数。
n 是一个非负整数，并且在32位有符号整型的范围内。

示例 1:
n = 5
硬币可排列成以下几行:
¤
¤ ¤
¤ ¤
因为第三行不完整，所以返回2.

示例 2:
n = 8
硬币可排列成以下几行:
¤
¤ ¤
¤ ¤ ¤
¤ ¤
因为第四行不完整，所以返回3.
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/arranging-coins

2. 思路分析：

假设可以形成k行完整的阶梯形状，那么需要满足1 + 2 + ... k = （1 + k）k / 2 <= n，我们求解这个方程即可，并且k需要满足是正整数，最终可以得到k <= (-1 + sqrt(1 + 8n) / 2，因为要使得k最大所以k = (-1 + sqrt(1 + 8n) / 2，我们在求解的时候向下取整即可。因为k是单调的所以还可以使用二分来解决。

3. 代码如下：

import math


class Solution:
    def arrangeCoins(self, n: int) -> int:
        return int((-1 + math.sqrt(1 + 8 * n)) / 2)

模拟：

class Solution:
    def arrangeCoins(self, n: int) -> int:
        i = 1
        s = 0
        while s + i <= n:
            s += i
            i += 1
        return i - 1

二分：

import math


class Solution:
    # 需要的硬币数与k是单调的所以可以使用二分
    def arrangeCoins(self, n: int) -> int:
        l, r = 0, n
        res = 0
        while l <= r:
            mid = l + r >> 1
            # 等差数列肯定是偶数不管mid是奇数还是偶数
            if (1 + mid) * mid // 2 > n:
                r = mid - 1
            else:
                l = mid + 1
        return l - 1