对新增的高度求和:H=(n+1)*n/2
H+3=x -> 1/2*n^2+1/2*n-(x-3)=0
对此二次方程求根取正数解:n=(2*x+1/4)^(1/2)-1/2
再向上取整得到结果.代码如下

import math
x=int(input())
print(math.ceil(math.sqrt(1.0/4.0+2.0*(x-3))-1.0/2.0))

a=int(input())
b=c=0
while(b<a-3):
c+=1
b=b+c
print(c)

挑战最短a.a

import math
print(math.ceil(math.sqrt(float(input())*2-5.75)-0.5))

m=int(input())#最后长度
n=3
d=0#天数
while n<m:
d=d+1
n=n+d
print(d)

m=int(input())
m>=3
n=0
a=(n+1)*n/2
while a<m-3:
n=n+1
a=(n+1)*n/2
print(n)