【刷穿LeetCode】441.排列硬币:「数学」&「

本文已参与「掘力星计划」,赢取创作大礼包,挑战创作激励金。

题目描述

这是 LeetCode 上的 441. 排列硬币 ,难度为 简单

Tag : 「数学」、「二分」

你总共有 n 枚硬币,并计划将它们按阶梯状排列。

对于一个由 k 行组成的阶梯,其第 i 行必须正好有 i 枚硬币。阶梯的最后一行可能是不完整的。

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

示例 1:

输入:n = 5

输出:2

解释:因为第三行不完整,所以返回 2 。
复制代码

示例 2:

输入:n = 8

输出:3

解释:因为第四行不完整,所以返回 3 。
复制代码

提示:


  • 1<=n<=23111 <= n <= 2^{31} - 1

数学

假设

nn

个硬币最多凑出

xx

行,根据等差数列求和公式

Sn=na1+n(n1)2dS_n = n * a_1 + \frac{n * (n - 1)}{2} * d

11

),可得

nn

xx

的关系:

x(x+1)2<=n\frac{x * (x + 1)}{2} <= n

问题为求满足上式前提下,

xx

的最大值为多少。

根据 求解一元二次方程 求根公式,可得:

x<=1+1+8n2x <= \left \lfloor \frac{- 1 + \sqrt{1 + 8 * n}}{2} \right \rfloor

代码:

class Solution {
    public int arrangeCoins(int n) {
        return (int)((Math.sqrt(1 + 8.0 * n) - 1) / 2);
    }
}
复制代码
  • 时间复杂度:
    O(1)O(1)

  • 空间复杂度:
    O(1)O(1)

二分

对于解方程不熟悉的同学,也可以使用「二分」来做。

对于

nn

个硬币而言,最多凑成的行数不会超过

nn

行,当且仅当

n=1n = 1

假设最大能够凑成的行数为

xx

,那么以

xx

为分割点的数轴上具有二段性,使用

nn

个硬币如果能够凑成

xx

行,那么必然也能够凑成少于等于

xx

行的阶梯硬币,必然不能凑成大于

xx

行的阶梯硬币。

对于

xx

行阶梯硬币所需要的硬币数量,还是可以利用等差数列求和公式直接算出。

代码:

class Solution {
    public int arrangeCoins(int n) {
        long l = 1, r = n;
        while (l < r) {
            long mid = l + r + 1 >> 1;
            if (mid * (mid + 1) / 2 <= n) l = mid;
            else r = mid - 1;
        }
        return (int)r;
    }
}
复制代码
  • 时间复杂度:
    O(logn)O(\log{n})

  • 空间复杂度:
    O(1)O(1)

最后

这是我们「刷穿 LeetCode」系列文章的第 No.441 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。

在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:github.com/SharingSour…

在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。