ccf 小明放学 python解法（100分）

 

if __name__=="__main__":

    r,y,g=map(int,input().split())
    ryg=r+y+g
    time =[0]*ryg
    for i in range(r):
        time[i]=r-i
    for i in range(r+g,ryg):
        time[i]=ryg-i+r
    print(time)
    n=int(input())
    tsum=0
    for i in range(n):
        flag,tt=map(int,input().split())
        if flag==0:
            tsum+=tt
            continue
        if flag==1:
            tt=r-tt                
        if flag==3:
            tt=r+g-tt
        if flag==2: 
            tt=ryg-tt               
        tp=(tt+tsum)%ryg
        tsum+=time[tp]
    print(tsum)

思路：红绿灯亮一轮的时间存入数组，按照红绿灯亮的顺序（“小明上学“中约定为红--> 绿-->黄-->红-->...），故[0,r ) 存放每轮红灯剩余等待时间 （r-i :倒计时），[ r , r+g ) 存放绿灯等待时间（0：直接通过），[ r+g , r+g+y ) 黄灯等待时间（r+y-i:黄灯亮完红灯亮），存完数组后，经过每个路口时我们只需要计算应该取出的数组位置。i 表示小明在某一轮的第i 秒到达路口（以每一轮红灯亮起的一瞬间为0时刻)

计算应该取出的数组位置：以输入 ”1 5“为例, 完整计算过程为：

  if flag==1:
         tt=r-tt

        tp=(tt+tsum)%ryg
        tsum+=time[tp]

因为输入的是红灯倒计时，所以小明相当于实在第30-5=25秒出发，当他到达该路口时，已经是 25+10=35秒了，所以此时查询time数组，得到对应等待时间。因为红绿灯是周期性亮灯，所以求余可以把tp约束到 [ 0, ryg )中。

输入 30 3 30则time数组是：

[30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 33, 32, 31]