Leetcode第441题：排列硬币

问题描述

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

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

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

解题思路

很快就可以想到连续数列的前n项和n*(n+1)/2。
n==5时：

i=1	i=2	i=3
n>i	n>i	n<i
n:=n-i=4	n:=n-i=2	return 2

n==8时：

i=1	i=2	i=3	i=4
n>i	n>i	n>i	n<i
n:=n-i=7	n:n-i=5	n:=n-i=2	return 3

C++代码

class Solution {
public:
    int arrangeCoins(int n) {
        if(n==0)
            return 0;
        else if(n==1)
            return 1;
            
        int i=1;
        while(n>i){
            //先排满第i行
            n-=i;
            //继续往下一行排
            ++i;
        }
        
        //剩余的可能刚好排满
        //也可能排不满
        if(n==i)
            return i;
        else
            return i-1;
    }
};

Python代码

class Solution:
    def arrangeCoins(self, n):
        if(n==0):
            return 0;
        elif(n==1):
            return 1;
        
        i=1
        while n>i:
            #先排第i行
            n=n-i
            #继续排下一行
            i=i+1
        
        #剩余的可能刚好排满第i行
        if(n==i):
            return i
        #也可能无法排满
        else:
            return i-1

(同样的算法思想，Python的执行效率这也太低了吧…)