[BZOJ2802] [Poi2012]Warehouse Store

题目描述

Description

有一家专卖一种商品的店,考虑连续的n天。
第i天上午会进货Ai件商品,中午的时候会有顾客需要购买Bi件商品,可以选择满足顾客的要求,或是无视掉他。
如果要满足顾客的需求,就必须要有足够的库存。问最多能够满足多少个顾客的需求。

Input

第一行一个正整数n (n<=250,000)。
第二行n个整数A1,A2,...An (0<=Ai<=10^9)。
第三行n个整数B1,B2,...Bn (0<=Bi<=10^9)。

Output

第一行一个正整数k,表示最多能满足k个顾客的需求。
第二行k个依次递增的正整数X1,X2,...,Xk,表示在第X1,X2,...,Xk天分别满足顾客的需求。

Sample Input

6
2 2 1 2 1 0
1 2 2 3 4 4

Sample Output

3
1 2 4

题目分析

我们维护一个大根堆。从第一天开始往后扫,假如当日剩余库存量足够满足当日的需求订单的话,就标记满足这个订单,并将这个订单中购买商品的数量加入大根堆。若当日剩余库存量不够满足当日的需求订单,且大根堆堆顶的订单大小比当日的订单大小大的话,就弹出堆顶,标记满足当日的订单,并取消掉堆顶的订单,将当日的订单大小加入堆中。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;
int n;
long long a[250010],b[250010];
int use[250010];
struct your
{
    long long c;
    int id;
};
struct cmp
{
     bool operator()(your x,your y)
    {
        return x.c<y.c;
    }
};
priority_queue<your,vector<your>,cmp> q;
long long now;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++) scanf("%lld",&b[i]);
    for(int i=1;i<=n;i++)
    {
        now+=a[i];
        if(now>=b[i])
        {
            your tmp;
            tmp.id=i,tmp.c=b[i],now-=b[i],q.push(tmp),use[i]=1;
        }
        else
        {
            if(q.size()&&q.top().c>b[i])
            {
                your tmp;
                tmp=q.top(),q.pop();
                now+=tmp.c,use[tmp.id]=0;
                tmp.id=i,tmp.c=b[i],now-=b[i],q.push(tmp),use[i]=1;
            }   
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++) ans+=(use[i]==1);
    printf("%d\n",ans);     
    for(int i=1;i<=n;i++) if(use[i]) printf("%d ",i);
    return 0;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注