【原题】
2802: [Poi2012]Warehouse Store
Time Limit: 10 Sec Memory Limit: 64 MBSec Special Judge
Submit: 94 Solved: 54
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
【分析】这道题我被坑的好久。
首先这肯定是贪心。我开始是这样想的：我们把顾客要求的多少排序，然后每次看看第I个顾客是否能给他，如果能的话就卖给他并更新我目前的钱。如果用f[i]表示到第i天剩余的钱的话，如果第K天的东西我卖了，我会把K+1~N的所有f元素都减去第K天顾客的要求数量。HHD表示应该减去1--K-1的f值。我们马上码好发现WA了。
【初始代码（开始用树状数组，后来怕写萎了，改成线段树了）】
#include<cstdio>
#include<algorithm>
#define L(x) (x&-x)
#define N 250005
using namespace std;
typedef long long ll;
ll f[N],data[N],write[N],temp,x,y,jia,n,i,ans,sum;
struct arr{ll x,id;}b[N];
struct arr2{ll l,r,sum,add;}a[N*4];
char opt[5];
inline void build(ll k,ll l,ll r)
{
a[k].add=0ll;a[k].l=l;a[k].r=r;
if (l==r) {a[k].sum=data[l];return;}
ll mid=(l+r)>>1;
build(k<<1,l,mid);build((k<<1)+1,mid+1,r);
a[k].sum=a[k<<1].sum+a[(k<<1)+1].sum;
}
inline void down(ll k)
{
if (a[k].add==0) return;
a[k<<1].sum+=a[k].add*(a[k<<1].r-a[k<<1].l+1ll);
a[(k<<1)+1].sum+=a[k].add*(a[(k<<1)+1].r-a[(k<<1)+1].l+1ll);
a[k<<1].add+=a[k].add;a[(k<<1)+1].add+=a[k].add;
a[k].add=0;
}
void update(ll k)
{
if (a[k].l>=x&&a[k].r<=y)
{
a[k].sum+=jia*1ll*(a[k].r-a[k].l+1ll);
a[k].add+=jia;return;
}
down(k);
ll mid=(a[k].l+a[k].r)>>1;
if (x<=mid) update(k<<1);
if (y>mid) update((k<<1)+1);
a[k].sum=a[k<<1].sum+a[(k<<1)+1].sum;
}
ll ask(ll k)
{
if (a[k].l>=x&&a[k].r<=y) return a[k].sum;
down(k);
ll mid=(a[k].l+a[k].r)>>1;ll o=0;
if (x<=mid) o+=ask(k<<1);
if (y>mid) o+=ask((k<<1)+1);
a[k].sum=a[k<<1].sum+a[(k<<1)+1].sum;
return o;
}
inline ll Read()
{
char ch=getchar();for (;ch<'0'||ch>'9';ch=getchar());
ll x=0;for (;ch>='0'&&ch<='9';ch=getchar()) x=x*10ll+ch-'0';
return x;
}
//inline void add(ll x,ll v){if (!x) return;for (;x<=n;x+=L(x)) f[x]+=v;}
//inline ll ask(ll x){ll s=0;for (;x;x-=L(x)) s+=f[x];return s;}
inline bool cmp(arr a,arr b){return a.x<b.x;}
int main()
{
n=Read();
for (i=1;i<=n;i++) x=Read(),data[i]=data[i-1]+x;
build(1,1,n);
for (i=1;i<=n;i++) b[i].x=Read(),b[i].id=i;
sort(b+1,b+n+1,cmp);
for (i=1;i<=n;i++)
{
x=y=i;temp=ask(1);
if (temp>=b[i].x) x=i,y=n,jia=-b[i].x,update(1),write[++ans]=b[i].id;
}
sort(write+1,write+ans+1);
printf("%lld\n",ans);
for (i=1;i<=ans;i++) printf("%lld ",write[i]);
return 0;
}
原来这样的更新是有问题的，无论是前面还是后面减，都有严重的类似后效性的东西。
经过SYC大神的指导，有一种完美的解决方案：从第一天开始每次判断当前的东西可不可以卖。如果可卖就卖掉并压入堆。如果卖不掉（缺货），去之前的堆中找最大值比较一下并决定是否卖。
【AC代码】
#include<cstdio>
#include<algorithm>
#include<queue>
#define N 250005
using namespace std;
typedef long long ll;
typedef pair<ll,ll>Pair;
Pair temp;
priority_queue<Pair>q;
inline ll Read()
{
char ch=getchar();for (;ch<'0'||ch>'9';ch=getchar());
ll x=0;for (;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
ll a[N],b[N],ans[N],now,n,i,big,id,cnt;
int main()
{
n=Read();
for (i=1;i<=n;i++) a[i]=Read();
for (i=1;i<=n;i++) b[i]=Read();
for (i=1;i<=n;i++)
{
now+=a[i];
if (now>=b[i]) now-=b[i],q.push(make_pair(b[i],i)),ans[i]=1;
else if (!q.empty())
{
temp=q.top();big=temp.first;id=temp.second;
if (big>b[i]) q.pop(),ans[id]=0,now+=(big-b[i]),q.push(make_pair(b[i],i)),ans[i]=1;
}
}
for (i=1;i<=n;i++) if (ans[i]) cnt++;
printf("%lld\n",cnt);
for (i=1;i<=n;i++) if (ans[i]) printf("%d ",i);
return 0;
}
bzoj 2802: [Poi2012]Warehouse Store 题解