poj3190 Stall Reservations(贪心+优先队列)

思路来源

https://blog.csdn.net/u014422052/article/details/44058525

题意

给N头牛,每头牛都对应一个挤奶区间,

一台挤奶机上的奶牛的挤奶区间不能相交,求最少需要几台挤奶机,

输出挤奶机和每头牛对应的挤奶机号。

题解

这题按现实生活模拟就好。

先来的先挤奶,为第一头开一台挤奶机。

后来的,如果和已经挤完了的最先结束的不冲突,就用最先结束的那台。

否则,就新开一台。

代码实现

将输入的按先来后到排序,

同from时间的结束时间早的对后续结果更有利。

pos记录原位置,便于恢复。

维护一个优先队列,按结束时间排序,栈顶为结束时间早的。

只要可以利用队顶结果就不用新开,否则就新开。

#include <iostream>
#include <algorithm> 
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <queue>
#include <bitset> 
const int INF=0x3f3f3f3f;
const int maxn=50005;
const int mod=1e9+7;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int> 
#define si set<int>
#define pii pair<int,int> 
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a)) 
using namespace std;
struct node
{
	int from,to,pos;
};
node p[maxn];
int ans[maxn];
bool operator<(node a,node b)
{
    return a.to>b.to;
}
bool cmp(node a,node b)//按开始时间排序 相同时结束时间小的入队 
{
	if(a.from!=b.from)return a.from<b.from;
	else return a.to<b.to;
} 
priority_queue<node>q;//优先队列里 结束时间小的先出 
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
	rep(i,0,n-1)sci(p[i].from),sci(p[i].to),p[i].pos=i;
	sort(p,p+n,cmp);
	int num=1;
	q.push(p[0]);
	ans[p[0].pos]=num;
	rep(i,1,n-1)
	{
		if(!q.empty()&&q.top().to<p[i].from)
		{
			ans[p[i].pos]=ans[q.top().pos];
			q.pop();//用已结束的暂时占用给当前牛,先结束的先拿来用 
		}
		else 
		{
			num++;
			ans[p[i].pos]=num;
		}
		q.push(p[i]);
	}
	printf("%d\n",num);
	rep(i,0,n-1)
	printf("%d\n",ans[i]);
    }
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Code92007/article/details/82917451