技術標籤：貪心貪心

小目錄

• • 連結
  • 題目描述
  • 樣例輸入
  • 樣例輸出
  • 思路
  • 程式碼

連結

YbtOJ高效進階 貪心-3

題目描述

有N頭牛在畜欄中吃草。每個畜欄在同一時間段只能提供給一頭牛吃草，所以可能會需要多個畜欄，給出第i頭牛開始吃草的時間區間 [ a i , b i ] [a_i, b_i] [ai​,bi​] ，求需要的最少畜欄數和每頭牛對應的畜欄方案。

樣例輸入

5
1 10
2 4
3 6
5 8
4 7

樣例輸出

4
1
2
3
2
4

思路

對每頭牛的開始時間從小到大排序
然後貪心？
對於第i個牛，它要麼是進新的畜欄，要麼是進已經沒牛的畜欄
1.進新的畜欄，那只有每個畜欄都有牛，才會開新的畜欄給這頭牛

程式碼

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>

using namespace std;

int n, ans[100005];
priority_queue <pair<int, int>, vector<pair<int, int> >
 
, greater<pair<int, int> > > Q;

struct tt 
{
	int l, r, num;
}a[100005];


bool cmp(tt a, tt b)
{
	return a.l < b.l;
}

int main()
{
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i)
		scanf("%d%d", &a[i].l, &a[i].r), a[i].num = i;
	sort(a + 1, a + n + 1
 
, cmp);
	Q.push(make_pair(a[1].r, 1));
	ans[a[1].num] = 1;
	for(int i = 2; i <= n; ++i)
	{
		if(!Q.size() || Q.top().first >= a[i].l)	
		{
			ans[a[i].num] = Q.size() + 1;
			Q.push(make_pair(a[i].r, Q.size() + 1));
		}//對應思路中的第一種情況
		else {
			ans[a[i].num] = Q.top().second; Q.pop();
			Q.push(make_pair(a[i].r, ans[a[i].num]));
		}//第二種情況
	} 
	printf("%d\n", Q.size());
	for(int i = 1; i <= n; ++i)
		printf("%d\n", ans[i]);
}