畜栏预定【贪心】【堆】

>Link

ybtoj畜栏预定


>解题思路

牛能否放进一个畜栏肯定要看这一个畜栏上一头牛的结束时间是否比现在这头牛开始吃草的时间晚
我一开始的思想是把每头牛按照开始吃草时间从小到大排序,记录现存的每个畜栏的最后一头牛结束吃草的时间,找到可以放进去的畜栏中上一次时间最早的,否则新建一个畜栏

但是实际上不需要这样,只要随便找一个可以放进去的畜栏就行了,因为放哪一个可以放的畜栏的效果都是一样的,我们已经把牛排了序,放进任何一个畜栏那个畜栏都会被更新为这头牛的结束时间,在处理下一头牛的时候,下一头牛的开始时间一定比当前牛的开始时间晚,本来就可以放进的话也一定可以被放进

所以我们对畜栏维护一个小根堆,每到一头牛就把它放进对顶的那个畜栏,如果放不进就新建畜栏


>代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define N 50010
using namespace std;

struct node
{
    
    
	int l, r, id;
} a[N];
struct tree
{
    
    
	int c, bh;
} t[N];
int n, cnt, ans[N];

bool cmp (node aa, node bb)
{
    
    
	if (aa.l != bb.l) return aa.l < bb.l;
	return aa.r < bb.r;
}
void up (int now)
{
    
    
	while (now > 1 && t[now / 2].c > t[now].c)
	{
    
    
		swap (t[now / 2], t[now]);
		now /= 2;
	}
}
void down (int now)
{
    
    
	int p;
	while ((now * 2 <= cnt && t[now * 2].c < t[now].c)
	      || (now * 2 + 1 <= cnt && t[now * 2 + 1].c < t[now].c))
	{
    
    
		p = now * 2;
		if (t[now * 2 + 1].c < t[now * 2].c) p++;
		swap (t[now], t[p]);
		now = p; 
	}
}

int main()
{
    
    
	scanf ("%d", &n);
	for (int i = 1; i <= n; i++)
	{
    
    
		scanf ("%d%d", &a[i].l, &a[i].r);
		a[i].id = i;
	}
	sort (a + 1, a + 1 + n, cmp);
	t[++cnt] = (tree){
    
    a[1].r, 1};
	ans[a[1].id] = 1;
	for (int i = 2; i <= n; i++)
	{
    
    
		if (a[i].l <= t[1].c)
		{
    
    
			t[++cnt] = (tree){
    
    a[i].r, cnt};
			ans[a[i].id] = cnt;
			up (cnt);
		}
		else
		{
    
    
			t[1].c = a[i].r;
			ans[a[i].id] = t[1].bh;
			down (1);
		}
	}
	printf ("%d\n", cnt);
	for (int i = 1; i <= n; i++)
	  printf ("%d\n", ans[i]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43010386/article/details/112138286