POJ 3091 Stall Reservation

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



你可能感兴趣的:(ACM,poj,ICPC,区间问题)