CCF 201709-2 公共钥匙盒

题目

代码

#include <iostream>
#include<algorithm>
#include <vector>
using namespace std;
bool cmp(int x,int y)
{
    return x>y;
}
int main(){
	//N个教室 K个老师（K行） 
	int N,K;
	cin>>N;
	cin>>K;
	//钥匙盒
	int key[N];
	for(int i=0;i<N;i++) key[i]=i+1;
	//借的钥匙 借的时间 还的时间
	int jiekey[K];
	int jietime[K];
	int huantime[K];
	//最大的时间
	int max=-1; 
	for(int i=0;i<K;i++){
		cin>>jiekey[i];
		cin>>jietime[i];
		cin>>huantime[i];
		huantime[i]+=jietime[i];
		if(huantime[i]>max) max=huantime[i];
	}
	//从时间开始遍历 时间从1开始 
	for(int i=1;i<=max;i++){
		//找当前时间点上是否有还行为 因为就算同时有借还 也是先还再借
		vector<int> temp; 
		for(int j=0;j<K;j++){
			//判断是否有还
			//因为需要考虑同时归还时钥匙从小到大还 所以先找到所有还的钥匙
			if(huantime[j]==i){
				temp.push_back(jiekey[j]);
			}
		}
		//向量排序
		sort(temp.begin(),temp.end(),cmp);
		//归还钥匙
		int index=0;
		while(!temp.empty()){
			if(key[index]==-1){
				key[index]=temp[temp.size()-1];
				temp.pop_back();
			}
			index++;
		}
		//找当前时间点上是否有借行为
		for(int j=0;j<K;j++){
			//是否有借
			if(jietime[j]==i){ //如果借走 把key里面当前位置的值赋为-1
				//借走的钥匙就是jiekey[j];
				for(int k=0;k<N;k++){
					if(key[k]==jiekey[j]){
						key[k]=-1;
						break;
					}
				}
			}
		}
	}
	//输出钥匙盒
	for(int i=0;i<N;i++) cout<<key[i]<<" "; 
	return 0;
}