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

 

啊……我要开始疯狂地写stl堆了

对于第i天的货物，它们只可能在第i到第n天用上

可以考虑倒序处理，每次把当前的b[i]扔进堆中，然后提取出当前堆中的最小b[k]，把a[i]给它。如果货物还有剩的就继续搞

因为倒序处理完序号有点乱还wa了一次……

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<ctime>
#define LL long long
#define inf 0x7ffffff
#define pa pair<int,int>
#define N 250010
using namespace std;
inline LL read()
{
    LL 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;
}
priority_queue<pa,vector<pa>,greater<pa> >q;
int n;
int a[N],b[N],ans;
bool num[N];
int main()
{
	n=read();
	for (int i=n;i>=1;i--)a[i]=read();
	for (int i=n;i>=1;i--)b[i]=read();
	while (!q.empty())q.pop();
	for (int i=1;i<=n;i++)
	{
		q.push(make_pair(b[i],n-i+1));
		while (a[i]&&!q.empty())
		{
			int f=q.top().first,rnk=q.top().second;
			q.pop();
			if (a[i]>=f)
			{
				num[rnk]=1;
				ans++;
				a[i]-=f;
			}else
			{
				q.push(make_pair(f-a[i],rnk));
				a[i]=0;
			}
		}
	}
	printf("%d\n",ans);
	for (int i=1;i<=n;i++)
	  if (num[i])printf("%d ",i);
	return 0;
}