[解题笔记] LeetCode 7月挑战题目之1

[解题笔记] LeetCode 7月挑战题目之1

这是我第一次参加 LeetCode (力扣) 的每月挑战题组，希望留下点笔记，大家可以参考和互相讨论。

第一天问：

假设你现在有 n 个硬币，然后你又那么无聊的想弄一段楼梯。这个楼梯要求是，在第 k 层必须有 k 个硬币 (第一层一个，第二层两个，如此类推)。

题目要求：已知 n 为非负整数而且是 32-位元带号整数 (32-bit signed integer)，求可堆砌的最大、最完整楼层。

例子 (引用自LeetCode)：

n = 5

The coins can form the following rows:
¤
¤ ¤
¤ ¤

Because the 3rd row is incomplete, we return 2.

解题思路 (1)：

首先，这个是一个标准的迭代问题，最简单的方法就是从 k = 1 开始，逐一递減 (k = 1, 2, 3, …)，直到 n 被減至 0 或者负数便可。

代码 (时间复杂度：O(n))

class Solution:
    def arrangeCoins(self, n: int) -> int:
    	for k in range(n + 1): #加1确保为正整数
            n = n - k # 负数，所以前一个 k已经是最完整
            if n < 0:
                return k - 1
            elif n == 0: #刚刚好，k就是答案
                return k

解题思路 (2)：

运用一点数学知识，我们就可以求出楼梯到第 k 层时所需要的硬币：
$$n=1+2+3+...+k=\frac{k(k+1)}{2}$$
所以，利用二分搜查法 (binary search) 原理，就可以把时间复杂度降为 O(log n)。那数学加法和二分搜查法有什么关系呢？

其实，如果把 0 和 n 分別当成二分搜查的最小数和最大数，然后在中间找一个 k 使得以上算式成立就可以。

代码 (时间复杂度：O(log n))

class Solution:
    def arrangeCoins(self, n: int) -> int:
    	left, right = 0, n
        while left <= right:
            k = (left + right) // 2
            if k * (k + 1) / 2 == n:
                return k
            elif k * (k + 1) / 2 > n:
                right = k - 1
            else:
                left = k + 1
        return right