leetcode 441.排列硬币

最新推荐文章于 2023-05-07 13:34:06 发布

黑暗主宰

最新推荐文章于 2023-05-07 13:34:06 发布

阅读量552

点赞数

分类专栏： leetcode刷题文章标签： leetcode 441 排列硬币二分法数学法迭代法

本文链接：https://blog.csdn.net/EngineerHe/article/details/103380203

版权

leetcode刷题专栏收录该内容

127 篇文章 2 订阅

订阅专栏

leetcode 441.排列硬币

题目描述

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

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

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

解题思路

方法一： 迭代法，直接看代码吧，很简单

class Solution {
public:
    int arrangeCoins(int n) {
        if(n<0){
            return -1;
        }
        int k = 1;
        while(n-k >= 0){
            n -= k;
            k++;
        }
        return k-1;
    }
};

方法二： 数学方法

可以推到一下数学公式
$\begin{array}{l} \frac{{x \times (x + 1)}}{2} = n\\ {(2x + 1)^2} = 8 \times n + 1\\ n = \frac{{\sqrt {8 \times n} - 1}}{2} \end{array}$
所以代码如下：

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

方法三： 二分法，使用二分法来缩小空间的访问

class Solution {
public:
    int arrangeCoins(int n) {
        long l = 0, r = n;
        long mid = l + (r-l)/2;

        while(l<r){
            if(mid*(mid+1) == (long)2*n){
                return mid;
            }
            else if(mid*(mid+1) < (long)2*n){
                l = mid+1;
            }
            else{
                r = mid;
            }
            mid = l + (r-l)/2;
        }
        return (l*(l+1)) == ((long)2*n) ? l: l-1;
    }
};