POJ 3190 Stall Reservations (贪心+优先队列)

Posted 2021-10-17 karshey

题
参考

排序排麻了…
这道题贪心+优先队列。
我们用一个优先队列来记录这个畜栏什么时候吃完，先吃完的排在前面。如果第一个畜栏的结束时间小于接下来牛的开始时间则可以共用。
在开始前我们要对牛进行排序，先开始的在前面，相同时间开始的就先结束的在前面，这样就保证最优。

关于优先队列的排序规则：重载小于号

struct node
{
	int l,r,i;
	bool operator<(const node &a)const
	{
		if(r==a.r) return l>a.l;
		else return r>a.r;
		//想要结束早的，如果一样早就要开始早的 
	}
}a[N];

其实没懂为什么要写l>a.l,但是可以输出看一下，符合就完事了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 

//
const int N=5e4+10;
struct node
{
	int l,r,i;
	bool operator<(const node &a)const
	{
		if(r==a.r) return l>a.l;
		else return r>a.r;
		//想要结束早的，如果一样早就要开始早的 
	}
}a[N];

bool cmp(node a,node b)
{
	if(a.l!=b.l) return a.l<b.l;
	else return a.r<b.r;
}
int anss[N];
int main()
{
	int n;cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>a[i].l>>a[i].r;
		a[i].i=i;
	}
	int ans=0;
	priority_queue<node>q;
	sort(a,a+n,cmp);
	for(int i=0;i<n;i++)
	{
		if(q.empty()) 
		{
			ans++;
			anss[a[i].i]=ans;			
		}
		else
		{
			//可以一起
			if(q.top().r<a[i].l)
			{
			//	cout<<q.top().r<<endl;
				anss[a[i].i]=anss[q.top().i];
				q.pop();				
			} 
			else
			{
				ans++;
				anss[a[i].i]=ans;				
			}			
		}
		q.push(a[i]);
	}
	cout<<ans<<endl;
	for(int i=0;i<n;i++) cout<<anss[i]<<endl;
	return 0; 
}

ps：牛的排序一定要先开始的在前面！