CSP 2018年12月 小明放学

CSP 201812-2 小明放学

我的CSP代码仓库：https://github.com/Joke-Lin/CSP

CSP题解目录：https://blog.csdn.net/qq_40953281/article/details/86035679

问题描述

试题编号：	201812-2
试题名称：	小明放学
时间限制：	1.0s
内存限制：	512.0MB
问题描述：	题目背景　　汉东省政法大学附属中学所在的光明区最近实施了名为“智慧光明”的智慧城市项目。具体到交通领域，通过“智慧光明”终端，可以看到光明区所有红绿灯此时此刻的状态。小明的学校也安装了“智慧光明”终端，小明想利用这个终端给出的信息，估算自己放学回到家的时间。问题描述　　一次放学的时候，小明已经规划好了自己回家的路线，并且能够预测经过各个路段的时间。同时，小明通过学校里安装的“智慧光明”终端，看到了出发时刻路上经过的所有红绿灯的指示状态。请帮忙计算小明此次回家所需要的时间。输入格式　　输入的第一行包含空格分隔的三个正整数 r、y、g，表示红绿灯的设置。这三个数均不超过 10^6。 　　输入的第二行包含一个正整数 n，表示小明总共经过的道路段数和路过的红绿灯数目。 　　接下来的 n 行，每行包含空格分隔的两个整数 k、t。k=0 表示经过了一段道路，将会耗时 t 秒，此处 t 不超过 10^6；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 ≤ 10^3。 　　 * 所有测试点保证 n ≤ 10^5。

题目本身并不难，和第一题就是需要单独重新判断以下当前的红绿灯状态。

题目中唯一需要注意的一点就是溢出判断，因为测试点规模为 10^5 * 10^6已经超出int类型的范围了

所以采取long long类型

还需要注意题目的红绿黄顺序，。。。迷

#include <iostream>
#include <cstdio>
#include <memory.h>

//#define LOCAL

using namespace std;
long long r,y,g;  // 1：红灯 2：黄灯 3：绿灯

// 输入红绿灯原来状态，以及时间，以及经过的时间，返回需要等待的时间
long long stayTime(long long state, long long timeTo, long long timeLast) {
    long long timeTemp;
    if(state == 1) timeTemp = r-timeTo;
    else if(state == 2) timeTemp = r+y+g - timeTo;
    else if(state == 3) timeTemp = r+g - timeTo;
    long long timeNow = (timeLast+timeTemp) % (r+g+y);
    if(timeNow < r) return r-timeNow;
    else if(timeNow >= r && timeNow < r+g) return 0;
    else if(timeNow >= r+g) return r+y+g - timeNow + r;
}

int main()
{
#ifdef LOCAL
    freopen("./in.txt","r",stdin);
    freopen("./out.txt","w",stdout);
#endif
    scanf("%lld%lld%lld",&r,&y,&g);
    long long res = 0;
    long long n; scanf("%lld",&n);
    for(long long i = 0;i < n;i++) {
        long long kind, clocks; scanf("%lld%lld",&kind,&clocks);
        if(kind == 0) {
            res += clocks;
            continue;
        }
        long long tempTime = stayTime(kind,clocks,res);
        res += tempTime;
    }
    printf("%lld",res);
    return 0;
}