挑战程序设计竞赛 2.2章习题 poj 3190 Stall Reservations(贪心+优先队列)

地址 https://vjudge.net/problem/POJ-3190

题目大意
一群母牛安排挤奶，挤奶需要占用一间牛棚。每头牛不能和其他母牛共用牛棚。
根据每头牛的挤奶时间安排牛棚，询问最少需要多少牛棚，并且给出每头牛挤奶使用的牛棚号。

输入
第一行一个数字N 表示有N头牛
下面N行为每头牛挤奶的起始时间 每行两个数字 空格间隔

输出
第一行输出最少需要多少牛栏
下面N行输出每头牛挤奶占用的牛棚编号

样例
Sample Input
5
1 10
2 4
3 6
5 8
4 7
Sample Output
4
1
2
3
2
4

解答
接收每头牛的挤奶起始时间，然后按照起始时间排序，这样就可以按照奶牛开始挤奶的时间进行牛棚的安排，如果有空的牛棚则直接使用，如果当前轮到某头牛挤奶，但是没有牛棚，则新添一间牛棚。
但是如何得到当前时间是否有空的牛棚。这里使用优先队列来存储牛棚，以占用牛棚的牛的挤奶结束时间进行排序，得到当前结束时间最早的牛棚。

题外话 考虑过使用哈希记录每头牛对应的牛棚号，但是poj不支持unordered_map
代码

// poj 3190.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

#include <iostream>
#include <queue>
#include <vector>
//#include <unordered_map>
#include <algorithm>

using namespace std;


/*
Input
Line 1: A single integer, N
Lines 2..N+1: Line i+1 describes cow i's milking interval with two space-separated integers.
Output
Line 1: The minimum number of stalls the barn must have.
Lines 2..N+1: Line i+1 describes the stall to 
which cow i will be assigned for her milking period.
Sample Input

5
1 10
2 4
3 6
5 8
4 7
Sample Output

4
1
2
3
2
4
*/


struct COW {
	int start;
	int end;
	int num;
	int stall;
};


struct cmp {
	bool operator()(struct COW& x, struct COW& y)
	{
		return  x.end > y.end;
	}
};

priority_queue<struct COW, vector<struct COW>, cmp> q;    //定义方法
int n;
 

bool cmp_COW(const struct COW& a, const struct COW& b) {
	return a.start < b.start;
}

struct COW mm[1000010];

int main()
{
	cin >> n;
	vector<struct COW> arr;
	for (int i = 0; i < n; i++) {
		int a, b;
		cin >> a >> b;

		struct COW t; t.start = a; t.end = b; t.num = i;
		arr.push_back(t);
	}

	sort(arr.begin(), arr.end(),cmp_COW);

	int ans = 0;
	//unordered_map<int, struct COW> mm;

	for (int i = 0; i < arr.size(); i++) {
		if (q.empty() || q.top().end >= arr[i].start) {
			ans++;
			arr[i].stall = ans;
			q.push(arr[i]);
		}
		else {
			struct COW t = q.top();
			mm[t.num] = t;
			q.pop();
			arr[i].stall = t.stall;
			q.push(arr[i]);
		}
	}
	
	cout << ans << endl;

	while (!q.empty()) {
		struct COW t = q.top();
		mm[t.num] = t;
		q.pop();
	}

	for (int i = 0; i < n; i++) {
		cout << mm[i].stall << endl;
	}
	
	return 0;
}