python基础运算

小李大学毕业开始工作，他想买一套现价为100万元的房子。若房子价格以每年百分之k（1≤k≤20）增长，小李的年薪是n（10≤n≤50）万元，假设小李未来的年薪不变，且每年所得的n万元全部积攒起来，请计算小李多少年能买下这套房子。如果超过20年小李也买不起这套房子，输出“很遗憾，小李买不起！”。

house = 100
money = 0
k = 2     # 自定义k
n = 20  # 自定义n
for year in range(1,22):
    if year>20:
        print("很遗憾，小李买不起！")
    else:
        house *= (1+(k/100))
        money += n
    if money > house:
        print(str(year)+"年")
        break

def buy_house(h,k,n):
    for y in range(1,21):
        house = h * (1+k/100)**y
        money = n * y
        if money > house:return f'小李{str(y)}年能买下这套房子'
    return '很遗憾，小李买不起！'

h = int(input('请输入房子现价：'))
k = int(input('请输入年利率：'))
n = int(input('请输入小李年薪：'))

print(buy_house(h,k,n))