ACMCODER-股神

有股神吗？
有，小赛就是！
经过严密的计算，小赛买了一支股票，他知道从他买股票的那天开始，股票会有以下变化：第一天不变，以后涨一天，跌一天，涨两天，跌一天，涨三天，跌一天...依此类推。
为方便计算，假设每次涨和跌皆为1，股票初始单价也为1，请计算买股票的第n天每股股票值多少钱？

 

 

代码如下：

#有股神吗？

def count_money(n):
    a = b = 1
    money = 1 #第一天的股票钱
    i = 2 #从第二天开始
    n = int(n)
    while i <= n:
        if a > 0:
            money += 1
            a -=1
        else:
            money -= 1
            b += 1
            a = b
        i+=1
    return money

res = []
money_list = []
import sys
try:
    while True:
        line1 = sys.stdin.readline().strip("\n")
        while line1 != '':
            res.append(line1)
            line1 = sys.stdin.readline().strip("\n")
        # print(res)
        for n in res:
            money = count_money(n)
            money_list.append(money)
        for j in money_list:
            print(j)
        break
except:
    pass

结果如下：

def count_money(n):
    a = b = 1
    money = 1
    i = 2
    n = int(n)
    while i <= n:
        if a > 0:
            money += 1
            a -=1
        else:
            money -= 1
            b += 1
            a = b
        i+=1
    return money

day=input()
if int(day) >=1 & int(day)<=10^9:
    print(count_money(day))
else:
    print('No')