CSP_小明放学

CSP_小明放学（数据结构上机第一次_4）

问题描述

一次放学的时候，小明已经规划好了自己回家的路线，并且能够预测经过各个路段的时间。同时，小明通过学校里安装的“智慧光明”终端，看到了出发时刻路上经过的所有红绿灯的指示状态。请帮忙计算小明此次回家所需要的时间。
输入格式：
　　输入的第一行包含空格分隔的三个正整数 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。

代码

#include<iostream>
#include<stdlib.h>
#include<malloc.h>
using namespace std;
static int sum=0;    //总时间 
int main()
{
	int r,y,g;    //红黄绿灯的周期(<=106)
	int n;    //经过道路和红绿灯的总数
	int k,t;    //k：道路0，红1，黄2，绿3    t：时间
	cin>>r>>y>>g;
	int cycle=r+y+g;    //一个周期时间 
	int add;    //每次增加的时间 
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>k>>t;
		if(k==0)    //道路 
		{
			sum+=t;
		}
		else if(k==1)    //红灯 
		{
			int sum1=sum;
			while(sum1>cycle)
				sum1-=cycle;    //减去周期 
			add=t-sum1;    //依然是红灯 
			if(add<0)
			{
				if(sum1<=t+g)    //红灯已变成绿灯 
					add=0;
				else if(sum1<=t+g+y)    //红灯已变成黄灯
					add=t+cycle-sum1;
				else if(sum1>t+g+y && sum1<=cycle)    //红灯再次变成红灯 
					add=t+cycle-sum1;		 
			}
			sum+=add;
		}
		else if(k==2)    //黄灯
		{
			int sum2=sum;
			while(sum2>cycle)
				sum2-=cycle;    //减去周期
			add=t-sum2;    //依然是黄灯 
			if(add<0)
			{
				if(sum2<=t+r)    //黄灯已变成红灯 
					add=t+r-sum2;
				else if(sum2<=t+r+g)    //黄灯已变成绿灯
					add=0;
				else if(sum2>t+r+g && sum2<=cycle)    //黄灯再次变成黄灯
					add=t+cycle+r-sum2;		 
			}
			sum+=add;
		}
		else if(k==3)    //绿灯
		{
			int sum3=sum;
			while(sum3>cycle)
				sum3-=cycle;    //减去周期
			if(t-sum3>0)    //依然是绿灯 
				add=0;
			else if(sum3<=t+y)    //绿灯已变成黄灯 
				add=t+y+r-sum3;
			else if(sum3<=t+r+y)    //绿灯已变称红灯
				add=t+y+r-sum3;	 
			else if(sum3>t+r+y && sum3<=cycle)    //绿灯再次变成绿灯
				add=0; 
			sum+=add; 
		}
	}
	cout<<sum;    //输出总时间 
}