leecode441：排列硬币（数学，二分查找）

leecode441：排列硬币（数学，二分查找）

441. 排列硬币

你总共有 n 枚硬币，你需要将它们摆成一个阶梯形状，第 k 行就必须正好有 k 枚硬币。

给定一个数字 n，找出可形成完整阶梯行的总行数。

n 是一个非负整数，并且在32位有符号整型的范围内。

示例 1:

n = 5

硬币可排列成以下几行:
¤
¤ ¤
¤ ¤

因为第三行不完整，所以返回2.

标签：二分查找，数学

我的题解：超时

class Solution {
    public int arrangeCoins(int n) {
        int result=0,i=0;
        while(result<=n)
        {
             i++;
            result+=i;
           
        }
        return i-1;
    }
}

大佬题解1:数学公式：

根据数学公式，k(k+1) /2 = n，可以得到其正数解为：k = sqrt(2n+1/4) - 1/2。然后求整即可。
唯一的问题是，这里2n+1/4有可能会超出sqrt函数的参数范围。
于是，我们可以变换一下， k = sqrt(2) * sqrt(n+1/8) - 1/2，这样求平方根就不会超限了。
于是，我们就有了这么一行代码

class Solution {
    public int arrangeCoins(int n) {
        return (int)(Math.sqrt(2) * Math.sqrt(n + 0.125) - 0.5);
    }
}

大佬题解二，三种思路

/**
     * 数学求解法 O(1) 不含求根预算
     *
     * @param n
     * @return
     */
    public static int arrangeCoins(int n) {
        return (int) (-1 + Math.sqrt(1 + 8 * (long) n)) / 2;
    }

    /**
     * 迭代求解法 O(n)//类似我的解法但是好像这些写不会 没超时
     *
     * @param n
     * @return
     */
    public static int arrangeCoins2(int n) {
        int i = 1;
        while (n >= i) {
            n -= i;
            i++;
        }
        return i - 1;
    }

    /**
     * 二分查找，O(log(n / 2 + 1))
     *
     * @param n
     * @return
     */
    public static int arrangeCoins3(int n) {
        long start = 0, end = (long) n / 2 + 1, mid;
        while (end - start > 1) {
            mid = (start + end) >>> 1;
            if (mid * (mid + 1) == (long) 2 * n) {
                return (int) mid;
            } else if (mid * (mid + 1) <= (long) 2 * n) {
                start = mid;
            } else {
                end = mid;
            }
        }
        return (end * (end + 1) == (long) 2 * n) ? (int) end : (int) start;
    }

@ElenoreChang 能讲解一下为啥要用long吗

因为判断条件 mid * (mid + 1) 用 int 会溢出的，当然也可以把变量类型设置为 int，判断条件中 if ((long)mid * (long)(mid + 1) == (long) 2 * n)