CSP201709-2公共钥匙盒

点击此处跳过题目描述及说明
点击此处跳转到源代码
问题描述
　　有一个学校的老师共用N个教室，按照规定，所有的钥匙都必须放在公共钥匙盒里，老师不能带钥匙回家。每次老师上课前，都从公共钥匙盒里找到自己上课的教室的钥匙去开门，上完课后，再将钥匙放回到钥匙盒中。
　　钥匙盒一共有N个挂钩，从左到右排成一排，用来挂N个教室的钥匙。一串钥匙没有固定的悬挂位置，但钥匙上有标识，所以老师们不会弄混钥匙。
　　每次取钥匙的时候，老师们都会找到自己所需要的钥匙将其取走，而不会移动其他钥匙。每次还钥匙的时候，还钥匙的老师会找到最左边的空的挂钩，将钥匙挂在这个挂钩上。如果有多位老师还钥匙，则他们按钥匙编号从小到大的顺序还。如果同一时刻既有老师还钥匙又有老师取钥匙，则老师们会先将钥匙全还回去再取出。
　　今天开始的时候钥匙是按编号从小到大的顺序放在钥匙盒里的。有K位老师要上课，给出每位老师所需要的钥匙、开始上课的时间和上课的时长，假设下课时间就是还钥匙时间，请问最终钥匙盒里面钥匙的顺序是怎样的？
输入格式
　　输入的第一行包含两个整数N, K。
　　接下来K行，每行三个整数w, s, c，分别表示一位老师要使用的钥匙编号、开始上课的时间和上课的时长。可能有多位老师使用同一把钥匙，但是老师使用钥匙的时间不会重叠。
　　保证输入数据满足输入格式，你不用检查数据合法性。
输出格式
　　输出一行，包含N个整数，相邻整数间用一个空格分隔，依次表示每个挂钩上挂的钥匙编号。
样例输入
5 2
4 3 3
2 2 7
样例输出
1 4 3 2 5
样例说明
　　第一位老师从时刻3开始使用4号教室的钥匙，使用3单位时间，所以在时刻6还钥匙。第二位老师从时刻2开始使用钥匙，使用7单位时间，所以在时刻9还钥匙。
　　每个关键时刻后的钥匙状态如下（X表示空）：
　　时刻2后为1X345；
　　时刻3后为1X3X5；
　　时刻6后为143X5；
　　时刻9后为14325。
样例输入
5 7
1 1 14
3 3 12
1 15 12
2 7 20
3 18 12
4 21 19
5 30 9
样例输出
1 2 3 5 4
评测用例规模与约定
　　对于30%的评测用例，1 ≤ N, K ≤ 10, 1 ≤ w ≤ N, 1 ≤ s, c ≤ 30；
　　对于60%的评测用例，1 ≤ N, K ≤ 50，1 ≤ w ≤ N，1 ≤ s ≤ 300，1 ≤ c ≤ 50；
　　对于所有评测用例，1 ≤ N, K ≤ 1000，1 ≤ w ≤ N，1 ≤ s ≤ 10000，1 ≤ c ≤ 100。

struct MOMENT {	//每一个结构体变量代表一个时刻
	int moment;				//记录是哪一个时刻
	vector<int>takeAway;	//存放当前时刻需要被借走的钥匙编号
	vector<int>giveBack;	//存放当前时刻要归还的钥匙编号
}m[2000];

把时刻定义成结构体，里面的成员有当前的时刻moment，当前时刻要借出的钥匙编号存放在vector takeAway中，当前时刻要归还的钥匙的编号存放在vector giveBack中。如果能正确的把输入的数据放到结构体中，然后再对结构体数组进行排序，那么模拟借钥匙和还钥匙的过程就会变得很简单。
思路分析：
对每个老师的w,s,c进行输入时，每输入一组，即处理一组。
定义一个整型变量count用来记录当前已有的时刻数，每当输入s和c之后，检查s这个时刻在结构体数组m中是否存在，若存在，则将w存入takeAway中；若不存在，则在数组中新加上s这个时刻，再将w存入takeAway中。时刻s+c是归还钥匙的时间，处理方法同借走钥匙的时间s。
数据全部输入完成之后，对结构体数组m按照成员变量moment递增的顺序排序。
排完序之后的结构体数组m，就可以用遍历的方法对其进行借还钥匙过程的模拟。
先模拟还钥匙过程，对挂钩进行遍历，若发现有空的挂钩，则将钥匙还回去。再模拟借钥匙过程，同样是对挂钩进行遍历，若发现某个挂钩上有自己需要的钥匙，则将其取下来。

#include<bits/stdc++.h>
using namespace std;
struct MOMENT {	//每一个结构体变量代表一个时刻
	int moment;				//记录是哪一个时刻
	vector<int>takeAway;	//存放当前时刻需要被借走的钥匙编号
	vector<int>giveBack;	//存放当前时刻要归还的钥匙编号
}m[2000];

bool compare(MOMENT s1, MOMENT s2) {
	return s1.moment < s2.moment;
}

int main()
{
	int N, K;	//N代表钥匙数，K代表需要借钥匙的老师数
	int w, s, c;	//w代表钥匙的编号，s代表借钥匙的时间，c代表使用时间
	int Pothook[1000];	//挂钩，因为最多有一千把钥匙，所以有一千个挂钩
	int count = 0;	//记录总共使用了多少个时刻
	cin >> N >> K;
	for (int i = 0; i < N; i++)	//给每个挂钩初始化一把钥匙
		Pothook[i] = i + 1;
	for (int i = 0; i < K; i++) {
		cin >> w >> s >> c;
		int j;
		//编号为w的钥匙在s时刻被借走
		for (j = 0; j < count; j++) 
			if (m[j].moment == s) {
				m[j].takeAway.push_back(w);
				break;
			}
			
		if (j == count) {
			m[count].moment = s;
			m[count].takeAway.push_back(w);
			count++;
		}
		//编号为w的钥匙在s+c时刻归还
		for (j = 0; j < count; j++) 
			if (m[j].moment == s + c) {
				m[j].giveBack.push_back(w);
				break;
			}

		if (j == count) {
			m[count].moment = s+c;
			m[count].giveBack.push_back(w);
			count++;
		}
	}
	sort(&m[0], &m[count],compare);	//对时刻进行升序排序
	for (int i = 0; i < count; i++) {
		int index = 0;
		sort(m[i].giveBack.begin(), m[i].giveBack.end());	//对要归还的钥匙编号进行排序
		//模拟还钥匙过程
		for (int j = 0; j < N && index < m[i].giveBack.size(); j++) {
			if (Pothook[j] == 0 ) {
				Pothook[j] = m[i].giveBack[index];
				index++;
			}
		}
		//模拟借钥匙过程
		for (int j = 0; j < m[i].takeAway.size(); j++) {
			for (int k = 0; k < N; k++)
				if (Pothook[k] == m[i].takeAway[j]) {
					Pothook[k] = 0;
					break;
				}
		}
	}
	for (int i = 0; i < N; i++)
		cout << Pothook[i]<<" ";
}

/*
5 2
4 3 3
2 2 7


5 7
1 1 14
3 3 12
1 15 12
2 7 20
3 18 12
4 21 19
5 30 9
*/
/*
在内存允许的情况下，数组的大小可以定义成实际需要的两倍
*/

注意事项：
刚开始我把数组m的大小定义成了1000，在csp系统上测试的时候总是显示运行错误，只有80分。最初以为是对vector的访问越界造成的，但看了好久也没发现有什么问题，而且如果是vector访问越界，应该得不到80分，后来发现是自己的结构体数组m的大小没定义对。我以为老师的数量K小于1000，那么发生借还钥匙的时刻也小于1000。忘了考虑极端情况，当老师数是1000，且所有老师的借钥匙时刻与还钥匙时刻都不相同，那么此时会有2000个时刻发生借钥匙或还钥匙操作。