【bzoj2802】[Poi2012]Warehouse Store 贪心+堆

贪心题真的好难呀！！！

按顺序枚举每个人
前面的方案在保证答案最大的前提下，库存尽量大
考虑新加进来的一个人，如果它买不到，看一看它前面选的有没有比它大的，如果有的话，把那个换成这个，答案不变，但库存变大了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<queue>
#define maxn 250010 

using namespace std;

int a[maxn],b[maxn],c[maxn];
priority_queue<pair<int,int> > q;
int n,m,ans,cnt;
long long sum=0;

int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	for (int i=1;i<=n;i++) scanf("%d",&b[i]);
	for (int i=1;i<=n;i++)
	{
		sum+=a[i];
		if (sum>=b[i]) ans++,sum-=b[i],q.push(make_pair(b[i],i));
		else if (!q.empty())
		{
			
			int t=q.top().first;
			if (t>b[i])
			{
				q.pop();
				sum-=b[i];sum+=t;
				q.push(make_pair(b[i],i));
			}
		}
	}
	printf("%d\n",ans);
	while (!q.empty()) c[++cnt]=q.top().second,q.pop();
	sort(c+1,c+cnt+1);
	for (int i=1;i<=ans;i++) printf("%d ",c[i]);printf("\n");
	return 0;
}