力扣：441. 排列硬币

题目：441. 排列硬币

难度： 简单

题目：

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

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

示例1

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

示例2

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

提示：

• 0 <= val <= 104
• 最多调用 addNum 和 getIntervals 方法 3 * 104 次

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/data-stream-as-disjoint-intervals/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

（1）模拟

其实题目的意思就是计算1 + 2 + 3 + … + k的和最接近n的那个k。所以模拟该过程，找到该k值。

class Solution {
public:
    int arrangeCoins(int n) {
        long long cnt = 0, k = 1, sum = 0;
        while (1) {
            sum += k++;
            if (sum <= n) {
                cnt++;
            } else {
                break;
            }
        }
        return cnt;
    }
};

（2）二分法

可以推出公式：k * (k + 1) / 2 = n，根据这个公式用二分法确定k，k通过二分不断逼近正确的结果。

class Solution {
public:
    int arrangeCoins(int n) {
        long l = 1, r = n;
        while (l <= r) {
            long mid = ((r - l) >> 1) + l;
            long target = mid * (mid + 1) / 2;
            if (n < target) {
                r = mid - 1;
            } else if (n > target) {
                l = mid + 1;
            } else {
                return mid;
            }
        }
        return r;
    }
};

（3）数学

由公式：k * (k + 1) / 2 = n，求出关于k的解。有两个解，除去一个负数，剩下：(-1 + sqrt(1 + 8 * n)) / 2的解即为所求。

注意：根号sqrt里的n输入是int，但是可能超出int，所以将其强转为long，以提高表达式的类型。

class Solution {
public:
    int arrangeCoins(int n) {
        return (int)(-1 + sqrt(1 + 8 * (long)n)) / 2;
    }
};