存款理财，程序怎么写

一年的时间，第一个礼拜存20，第二个礼拜存40，第三个礼拜存60，往后一个礼拜递增20，时间为一年，一共存入多少，程序怎么写，

# 一年的时间，第一个礼拜存20，第二个礼拜存40，第三个礼拜存60，往后一个礼拜递增20，时间为一年，一共存入多少?
import math

year = 365  # 设一年的天数为365天
t = 20  # 定义递增值为20
d = 7  # 定义一个礼拜为7天
s = 0  # 定义一年存入
z = math.ceil(year / d)  # 计算一年内多少个礼拜，不满一个礼拜按一个礼拜计算，53
for i in range(0, z):
    if i + 1 == 1:
        s = t * (i + 1)
    else:
        s = s + t * (i + 1)
print(s)

麻烦点个采纳