POJ 3091题目大意如下:
农场主有N头牛,每头牛对于自己的供奶时间非常挑剔,只在固定时间内的某个空的隔间供奶。但是农场主有不能无限制的给每头牛分配供奶隔间,所以需要通过设计程序来求出一个最优的方案,使得使用最少隔间满足所有奶牛的要求。
这是一个区间问题,问题给定每头牛的产奶时间段,首先可以根据每头牛的产奶时间段的下限优先排序(如果下限相同,再考虑上限)。之后在寻找最优方案的过程中,则可以使用优先队列按照上限优先(因为要使得每个隔间得到充分利用,所以尽量使得上限小的在前,而且如果最小上限都不能满足放在同一组的条件,那么之后的就可以不看,直接创建一个新的组)排序入队。过程中再与之后还未添加入队列,而即将添加入队列的的区间元素的下限作比较,如果满足下限大于当前出队元素的上限,则说明这两个元素可以放在同一个组里,同时这里有个优化处理,就是使用了一个flag数组针对每个区间纪录一个组别,但是在问题处理过程中,又实际不去看flag,因为问题关注的点是:一个区间能不能跟在另一个区间的后面。复杂度为O(n),下面是代码:
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 50000;
struct Section{
int left, right;
int index;
}cow[maxn + 1];
int N;
int flag[maxn];
struct Camp {
bool operator () (const Section& lhs, const Section& rhs) {
return lhs.right > rhs.right || (lhs.right == rhs.right && lhs.left > rhs.left);
}
};
bool comp(const Section& lhs, const Section& rhs) {
return lhs.left < rhs.left || (lhs.left == rhs.left && lhs.right < rhs.right);
}
void solve() {
int totalGroup = 0;
//按照上限优先排序
priority_queue<Section, vector<Section>, Camp> que;
for (int i = 0; i < N; i++) {
if (!que.empty() && que.top().right < cow[i].left) {
flag[cow[i].index] = flag[que.top().index];
que.pop();
}
else flag[cow[i].index] = ++totalGroup;
que.push(cow[i]);
}
printf("%d\n", totalGroup);
for (int i = 0; i < N; i++) printf("%d\n", flag[i]);
}
int main(int argc, const char * argv[]) {
// insert code here...
scanf("%d", &N);
for (int i = 0; i < N; i++) {
scanf("%d %d", &cow[i].left, &cow[i].right);
cow[i].index = i;
}
//按照下限优先排序
sort(cow, cow + N, comp);
solve();
return 0;
}