CSP-小明放学Python实现(有注释)

题目背景
　　汉东省政法大学附属中学所在的光明区最近实施了名为“智慧光明”的智慧城市项目。具体到交通领域，通过“智慧光明”终端，可以看到光明区所有红绿灯此时此刻的状态。小明的学校也安装了“智慧光明”终端，小明想利用这个终端给出的信息，估算自己放学回到家的时间。
问题描述
　　一次放学的时候，小明已经规划好了自己回家的路线，并且能够预测经过各个路段的时间。同时，小明通过学校里安装的“智慧光明”终端，看到了出发时刻路上经过的所有红绿灯的指示状态。请帮忙计算小明此次回家所需要的时间。
输入格式
　　输入的第一行包含空格分隔的三个正整数 r、y、g，表示红绿灯的设置。这三个数均不超过 106。
　　输入的第二行包含一个正整数 n，表示小明总共经过的道路段数和路过的红绿灯数目。
　　接下来的 n 行，每行包含空格分隔的两个整数 k、t。k=0 表示经过了一段道路，将会耗时 t 秒，此处 t 不超过 106；k=1、2、3 时，分别表示出发时刻，此处的红绿灯状态是红灯、黄灯、绿灯，且倒计时显示牌上显示的数字是 t，此处 t 分别不会超过 r、y、g。
输出格式
　　输出一个数字，表示此次小明放学回家所用的时间。
样例输入
30 3 30
8
0 10
1 5
0 11
2 2
0 6
0 3
3 10
0 3
样例输出
46
样例说明
　　小明先经过第一段路，用时 10 秒。第一盏红绿灯出发时是红灯，还剩 5 秒；小明到达路口时，这个红绿灯已经变为绿灯，不用等待直接通过。接下来经过第二段路，用时 11 秒。第二盏红绿灯出发时是黄灯，还剩两秒；小明到达路口时，这个红绿灯已经变为红灯，还剩 11 秒。接下来经过第三、第四段路，用时 9 秒。第三盏红绿灯出发时是绿灯，还剩 10 秒；小明到达路口时，这个红绿灯已经变为红灯，还剩两秒。接下来经过最后一段路，用时 3 秒。共计 10+11+11+9+2+3 = 46 秒。
评测用例规模与约定
　　有些测试点具有特殊的性质：
　　* 前 2 个测试点中不存在任何信号灯。
　　测试点的输入数据规模：
　　* 前 6 个测试点保证 n ≤ 103。
　　* 所有测试点保证 n ≤ 105。
思路：设置一个变量sum_time记录之前总共花费的时间，遇到0路段直接把时间累加到sum_time，然后每次遇到非0的路段都根据这个路口出发时的灯的状态计算当前时刻这个灯的状态，下面我做了个草图举个例子。
r,g,y = 30,30,3，假设前面已经经过好多路段了，然后当前遇到的这个路口出发时是红灯还剩5秒的状态（代码中我用start_time这个变量表示这个时间）,设置一个temp变量，temp=sum_time%(r+g+y)，（假如temp=0就说明之前经过的时间循环了好多次之后又刚好到了和出发时一样的状态了），temp的取值范围如下图蓝色部分所示，最下面的黄色轴为x的范围，x = temp-start_time，x的取值所在的颜色区间就对应的上面那个坐标轴中的颜色。

r,y,g = map(int,input().split())
n = int(input())
nums = []
for i in range(n):
    temp = list(map(int,input().split()))
    nums.append(temp)
# 记录总时间
sum_times = 0
for index in nums:
    start_status = index[0]  # 出发时此处的状态
    start_time = index[1]  # 出发时该状态剩余的时间
    # 直接通过
    if start_status == 0:
        sum_times += start_time
    # 如果出发时此地是红灯
    elif start_status == 1:
        temp = sum_times%(r+y+g)
        x = temp-start_time
        if x <= 0:  # 说明还是红灯,还得等待
            sum_times += abs(x)
        elif x < g: # 说明是绿灯可以直接通行
            sum_times += 0
        elif x < g+y:  # 说明是黄灯
            sum_times += (g+y-x)+r
        else:   # 下一轮的红灯
            sum_times += r-(x-g-y)
    # 出发时刻此地是黄灯
    elif start_status == 2:
        temp = sum_times%(r+y+g)
        x = temp-start_time
        if x <= 0:  # 说明还是黄灯,还得等待
            sum_times += abs(x)+r
        elif x < r: # 说明是红灯
            sum_times += r-x
        elif x < r+g:  # 说明是绿灯
            sum_times += 0
        else:   # 下一轮的黄灯
            sum_times += y-(x-g-r)+r
    # 出发时刻此地是绿灯
    else:
        temp = sum_times%(r+y+g)
        x = temp-start_time
        if x <= 0:  # 说明还是绿灯
            sum_times += 0
        elif x < y: # 说明是黄灯,减去需要等待的黄灯时间再加上红灯的时间
            sum_times += (y-x)+r
        elif x < y+r:  # 说明是红灯
            sum_times += y+r-x
        else:   # 下一轮的绿灯
            sum_times += 0
print(sum_times)