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

来源:互联网 发布:高频彩票算法大揭秘 编辑:程序博客网 时间:2024/06/03 16:19

贪心题真的好难呀!!!

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


#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;}


0 0