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;
}