【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<bits/stdc++.h>
#define pa pair<int,int>
typedef long long ll;
const int N=10000000;
const int mod=1000000007;
const double eps=0.00000001;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,a[250005],b[250005],tot;
ll sum;
priority_queue<pa,vector<pa> >q;
int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++) b[i]=read();
    for (int i=1;i<=n;i++)
    {
        sum+=a[i];
        if (sum>=b[i])
        {
            sum-=b[i];
            q.push(make_pair(b[i],i));
        }
        else if (!q.empty()) if (q.top().first>b[i])
        {
            sum=sum+q.top().first-b[i];
            q.pop();
            q.push(make_pair(b[i],i));
        }
    }
    printf("%d\n",q.size());
    while (!q.empty()){a[++tot]=q.top().second;q.pop();}
    sort(a+1,a+tot+1);
    for (int i=1;i<=tot;i++) printf("%d ",a[i]);
    return 0;
}