leetcode 441. 排列硬币(Arranging Coins)

题目描述:

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

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

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

示例 1:

n = 5

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

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

示例 2:

n = 8

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

因为第四行不完整,所以返回3.

解法:

class Solution {
public:
    int arrangeCoins(int n) {
        // (l + 1)*l/2 <= n < (l+2)*(l+1)/2
        long long tmp = n;
        tmp *= 2;
        long long res = sqrt(tmp);    // l <= res < l+2
        if((res+1)*res/2 > n){
            return res - 1;
        }else{
            return res;
        }
    }
};

猜你喜欢

转载自www.cnblogs.com/zhanzq/p/10583392.html