CSP 201709-2 公共钥匙盒

vis[]存储每个位置的钥匙编号
a[]储存每个钥匙的位置
注意：cmp1先后比较顺序是时间、类型再是编号，刚开始搞错顺序调整后满分。

#include<bits/stdc++.h>
#define maxnn 1000005
using namespace std;
typedef long long ll;
int vis[1005];//存储每个位置的钥匙编号 
int a[1005];//储存每个钥匙的位置 
struct key{
	int keyid;
	int time;//同时代表取出时间和归还时间 
	int type;//取出为0 归还为1 
};
struct cmp1
{
	bool operator () (const key &a1, const key &a2)		  
	{
        if(a1.time!=a2.time) return a1.time>a2.time;//首先判断时间，按照时间从小到大进行排序
        if(a1.type!=a2.type)return a1.type<a2.type;//再判断类型，按照还钥匙优先排序
        return a1.keyid>a2.keyid;//再判断编号，按照编号从小到大进行排序
	}
};
priority_queue<key, vector<key>,cmp1 > q;
priority_queue<int,vector<int>,greater<int> >p;//递增顺序存储挂钩空出的位置 
void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		vis[i] = i;
		a[i]=i;
	}
}
int main()
{
	int N,K;
	int w,s,c;
	cin>>N>>K;
	init(N);
	for(int i=0;i<K;i++)
	{
		key a,b;
		cin>>w>>s>>c;
		a.keyid=w;//w编号钥匙取出时刻 
		a.time=s;
		a.type=0;
		b.keyid=w;//w编号钥匙归还时刻 
		b.time=s+c;
		b.type=1;
		q.push(a);
		q.push(b);

	}
	while(!q.empty())
	{
		key x = q.top();
		q.pop();
		if(x.type==0)//取出操作 
		{
			if(a[x.keyid])//尚未被取出 
			{
				vis[a[x.keyid]]=0;//取出标记 
				p.push(a[x.keyid]);//空位置压入队列 
			}
			a[x.keyid] = 0;
		}
		else if(x.type==1)//归还操作 
		{
			int pos = p.top();
			p.pop();
			vis[pos] = x.keyid;//从左到右第一个空位放置钥匙 
			a[x.keyid] = pos;//更新钥匙位置 
			
		}
	}
	for(int i=1;i<=N;i++)
	{
		cout<<vis[i]<<' ';
	}
	return 0;
}