Python 三天打鱼两天晒网问题

郭、王两位大侠同上终南山习武，两人最初的战力值均为100。王大侠骨骼清奇，天赋较高，每练功一天，战力增加2‰，郭大侠比较愚笨，每练功一天，战力增加1‰。如果休假一天不练，两人的战力均减少1‰。相较于王大侠，郭大侠更加勤奋，日日练功，从不休息，而王大侠，则三天打渔，两天晒网，也就是每5天的前三天练功，后两天休假。

请编写程序，计算N天之后，郭大侠和王大侠战力分别是多少？

N = int(input())
guo = wang = 100
i = 1
while i <= N:
    guo *= 1.001
    if 0 < i%5 < 4: wang *= 1.002
    else: wang *= 0.999
    i += 1
print(guo)
print(wang)

def w(n):
    gl=100
    t=n%5
    z=n//5*0.004


    if t<=3:

        z+=0.002*t
    else:
        z += 0.006-0.001*t

    return gl*(1+z)
def g(n):
    gl=100
    gl=gl*(1+0.001*n)
    return gl

print(w(104),g(104))