leetcode441. 排列硬币(python)

题目链接

题目描述：

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

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

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

示例 1:

n = 5

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

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

n = 8

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

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

解题思路：

方法1：
时间超长，算出第k行需要的硬币直到所需硬币大于等于n为止

class Solution(object):
    def arrangeCoins(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n==1 or n==0:return n
        for i in range(2,100000):
            x=(1+i)*i//2
            if x==n:return i
            if x>n:return i-1