我如何改善/简化“上拉序列累加器”？

Question

我想创建一个描述体育游戏“ Leader”的功能。这样做的目的是使俯卧撑尽可能多，每次重复可以增加1，而当达到最大次数时，下一次重复将减少1，直到最终达到0个俯卧撑。

我设法使用字典来做到这一点，但是我认为可以用更容易的方式做到这一点。

def leader_step(max):
    i = 0
    psps = {0: 0}
    k = 1
    cnt = {0:0}
    while max >= psps[i]:
        if psps[i] == max:      # decrease push-ups as they reach max
            k = -1
        i += 1
        psps[i] = k + psps[i-1]
        if psps[i] < 1:         # game stops when you reach 1 push-up
            del psps[i]
            break
        cnt[i] = cnt[i - 1] + psps[i]
    del psps[0]
    del cnt[0]
    return psps.values(), cnt.values()

函数应返回2个序列： 1）显示每次重复的俯卧撑次数 2）显示每次重复进行俯卧撑的总次数