首页 > 代码库 > BZOJ 2802 Poi2012 Warehouse Store 堆+贪心

BZOJ 2802 Poi2012 Warehouse Store 堆+贪心

题目大意:有n天,早上进货中午卖,可以选择卖或者不卖,问最多可以卖出多少人

首先贪心的思想是如果当前能卖就卖

但是这样不一定是最优的

比如说我第一天来一个人把所有的库存都买走了 然后后面基本没有补给 后面的人都饿死了

因此我们维护一个大根堆来记录我们都卖出了多少份

如果有一个人买不到 我们去大根堆里寻找有没有买的比他多的 如果有 把之前的人取消 卖给这个人

这样虽然不能增加答案 但是可以增加库存

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 250100
using namespace std;
typedef pair<int,int> abcd;
int n,a[M],b[M],ans[M];
long long stock;
namespace Priority_Heap{
	abcd heap[M];int top;
	void Insert(abcd x)
	{
		heap[++top]=x;
		int t=top;
		while(t>1)
		{
			if(heap[t]>heap[t>>1])
				swap(heap[t],heap[t>>1]),t>>=1;
			else
				break;
		}
	}
	void Pop()
	{
		heap[1]=heap[top--];
		int t=2;
		while(t<=top)
		{
			if(t<top&&heap[t+1]>heap[t])
				++t;
			if(heap[t]>heap[t>>1])
				swap(heap[t],heap[t>>1]),t<<=1;
			else
				break;
		}
	}
}
int main()
{
	int i;
	cin>>n;
	for(i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
		scanf("%d",&b[i]);
	for(i=1;i<=n;i++)
	{
		stock+=a[i];
		if(stock>=b[i])
			stock-=b[i],Priority_Heap::Insert(abcd(b[i],i));
		else
		{
			if(Priority_Heap::heap[1].first<b[i])
				continue;
			stock+=Priority_Heap::heap[1].first;
			stock-=b[i];
			Priority_Heap::Pop();
			Priority_Heap::Insert(abcd(b[i],i));
		}
	}
	cout<<Priority_Heap::top<<endl;
	for(i=1;i<=Priority_Heap::top;i++)
		ans[++ans[0]]=Priority_Heap::heap[i].second;
	sort(ans+1,ans+ans[0]+1);
	for(i=1;i<=ans[0];i++)
		printf("%d%c",ans[i],i==ans[0]?'\n':' ');
	return 0;
}


BZOJ 2802 Poi2012 Warehouse Store 堆+贪心