POJ 3190 Stall Reservations

http://poj.org/problem?id=3190

//最初思路 按数组排序 也好 堆维护也好 都是想 让开始时间 > 结束时间的牛合并 然后最后求出 个数就好
//但是这样无法求得每头牛的分配过程
//所以 模拟 加贪心 堆模拟正在挤奶的机器 按结束时间从小到大排列 没搜索加入一头牛 就结束一头牛的挤奶
//这样可以求得每头牛 在哪个机器挤奶 并且如果 新加入牛时 这头牛已经结束 那么它就可以接着使用 否则 只能新加一台机器
//然后stall就是 进一头牛 确定一个的stall 这样就是纯模拟这个个过程
//一道贪心、模拟的好题

 

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <queue>

using namespace std;

int N;
int stall[50003];
struct Section
{
    int x,y;
    int pos;
    bool operator < (Section a) const //堆按结束时间升序排列
    {
        if (y == a.y) return x > a.x;
        else return y > a.y;
    }

}section[50003];

bool cmp(Section a, Section b)//数组按开始时间升序排列
{
    if(a.x == b.x) return a.y < b.y;
    else return a.x < b.x;
}

priority_queue<Section> que;
int solve()
{
    int ans = 0;
    Section node;
    for (int i = 0; i < N;i ++)
    {
        if ( !que.empty() )
        {
            node = que.top();
            if (que.top().y < section[i].x)//如果加进这个section的时候有结束的奶牛-->>>这里错了 不是当结束时间和开始时间相等时不能接替的
            {
                stall[section[i].pos] = stall[que.top().pos];
                que.pop();//这头奶牛就可以出去了
            }
            else
            {
                ans++;
                stall[section[i].pos] = ans;
            }
        }
        else
        {
            ans++;
            stall[section[i].pos] = ans;
        }
        que.push(section[i]);
    }
    return ans;
}

int main()
{
    freopen("in.txt", "r", stdin);
    while (~scanf("%d", &N))
    {
        while (!que.empty()) que.pop();
        for (int i = 0; i < N; i++)
        {
            scanf("%d%d", &section[i].x, &section[i].y);
            section[i].pos = i;
            //stall[section[i].pos] = i;//安排最初的stall
        }
        sort(section, section+N, cmp);//当自定义比较函数时 是可以忽略 结构体内部的操作符重载的
        int ans = solve();
        printf("%d\n",ans);
        for (int i = 0;i < N; i++)
        {
            printf("%d\n", stall[i]);
        }
    }
    return 0;
}