Python 历届真题 杨辉三角形【第十二届】【省赛】【研究生组】

Python 历届真题 杨辉三角形【第十二届】【省赛】【研究生组】

暴力算法只有40分

import sys
n = int(input())
if n==1:
    print(1)
    sys.exit()
yh = [1,1]
yh2 = []
for i in range(3,100):
    yh2.append(1)
    for j in range(2,i):
        s = yh[j-2]+yh[j-1]
        yh2.append(s)
        if s==n:
            print(int(0.5*(i-1)*(i-1)+0.5*(i-1)+j))
            sys.exit()      
    yh2.append(1)
    yh = yh2
    yh2 = []