CCF 201709-2 公共钥匙盒 100分

Posted 2021-07-11 登登登ccc

1、vector+结构体万能模板

2、使用构造函数对结构体赋值更方便

3、将开始和结束两个时间点作为不同结构体存入数组，以便排序

#include <bits/stdc++.h>

using namespace std;

struct Key {
	int room;
	int time;
	int operators; // 0:放回, 1:取
	Key(int room1,int time1,int operators1) { //构造函数
		room = room1;
		time = time1;
		operators = operators1;
	}
};

bool cmp(Key a,Key b) {
	if(a.time < b.time) return true;
	else if(a.time == b.time && a.operators < b.operators) return true;
	else if(a.time == b.time && a.operators == b.operators && a.room < b.room) return true;
	return false;//首先根据时间的先后排序。时间相同，根据取放排序，先放后取。时间相同，是相同的操作，先房间号低的操作。
}

int main() {
	int N, K;
	cin >> N >> K;
	vector<Key> v;
	vector<int> res(N+1);//比N大一防止向量数组溢出，N+1个参数
	for(int n=1; n<=N; n++)
		res[n] = n;
	for(int k=0; k<K; k++) {
		int room, begin, length;
		cin >> room >> begin >> length;
		v.push_back(Key(room, begin, 1));//取钥匙输入开始时间
		v.push_back(Key(room, begin+length, 0));//放回钥匙输入结束时间（开始时间+持续时间）
	}
	sort(v.begin(), v.end(),cmp);//调用cmp函数，给sort()排序加上具体的规则
	for(int i=0; i<v.size(); i++) {
		if( v[i].operators == 0) { // 放回
			for(int n=1; n<=N; n++) {
				if(res[n] == -1) {
					res[n] = v[i].room;
					break;
				}
			}
		} else { // 取钥匙
			for(int n=1; n<=N; n++) {
				if(res[n] == v[i].room) {
					res[n] = -1;//-1是对拿走钥匙后对空位置的标记
					break;
				}
			}
		}
	}
	for(int n=1; n<=N; n++) {
		cout<< res[n] << " ";
	}
	return 0;
}

转载自 CCF 201709-02公共钥匙盒_u012585224的专栏-CSDN博客