码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj 2802: [Poi2012]Warehouse Store 题解

时间:2014-06-05 08:07:48      阅读:243      评论:0      收藏:0      [点我收藏+]

标签:题解   bzoj   poi   贪心      

【原题】

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 题解,布布扣,bubuko.com

bzoj 2802: [Poi2012]Warehouse Store 题解

标签:题解   bzoj   poi   贪心      

原文地址:http://blog.csdn.net/jiangshibiao/article/details/27235807

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!