题目背景】
汉东省政法大学附属中学所在的光明区最近实施了名为“智慧光明”的智慧城市项目。具体到交通领域,通过“智慧光明”终端,可以看到光明区所有红绿灯此时此刻的状态。小明的学校也安装了“智慧光明”终端,小明想利用这个终端给出的信息,估算自己放学回到家的时间。
【问题描述】
一次放学的时候,小明已经规划好了自己回家的路线,并且能够预测经过各个路段的时间。同时,小明通过学校里安装的“智慧光明”终端,看到了出发时刻路上经过的所有红绿灯的指示状态。请帮忙计算小明此次回家所需要的时间。
【输入形式】
输入的第一行包含空格分隔的三个正整数 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 秒。
题解:
#include <iostream>
using namespace std;
long long r, y, g,ans = 0;
int n;
long long remain,period,mod;
int main() {
cin >> r >> y >> g;
cin >> n;
for (int i = 0; i < n; ++i) {
int k, t;
cin >> k >> t;
if (k == 0) {
ans += t;
} else {
long long time = ans;
if (k == 1) { // 初始是红灯
if (time <= t) {
ans += t - time;
} else {
remain = time - t;
period = g + y + r;
mod = remain % period;
if (mod < g) {
// 绿灯,无需等待
} else if (mod < g + y) {
ans += (g + y - mod) + r;
} else {
ans += (g + y + r - mod);
}
}
} else if (k == 2) { // 初始是黄灯
if (time <= t) {
ans += t - time + r; // 黄灯之后是红灯
} else {
remain = time - t;
period = r + g + y;
mod = remain % period;
if (mod < r) {
ans += (r - mod);
} else if (mod < r + g) {
// 绿灯,无需等待
} else {
ans += ( (r + g + y - mod) ) + r;
}
}
} else if (k == 3) { // 初始是绿灯
if (time > t) {
remain = time - t;
period = y + r + g;
mod = remain % period;
if (mod < y) {
ans += (y - mod) + r;
} else if (mod < y + r) {
ans += (y + r - mod);
}
}
}
}
}
cout << ans << endl;
return 0;
}