【BZOJ 2802】 2802: [Poi2012]Warehouse Store （贪心）

2802: [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

HINT

Source

鸣谢Oimaster

【分析】

　　可爱的贪心题。

　　【并没有秒，并且表示想了很久，并且表示样例调了挺久】

　　【幸好1A】【好吧学不会大颓果的来者不拒思想ORZ】

　　先说我的方法：

　　每次选最小的bi取，然后库存减掉，树状数组维护前缀和。

　　减库存的时候从当前位置往前减，减到0为止，不过这样会很慢，我用个链表维护，如果是0就直接跳过去了。

　　【实测不用链表真心TLE。。

　　

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<set>
 using namespace std;
 #define Maxn 250010
 #define LL long long

 int a[Maxn],b[Maxn],c[Maxn];
 LL d[Maxn];
 int lt[Maxn],nt[Maxn];

 bool cmp(int x,int y)
 {
     if(b[x]==b[y]) return x>y;
     return b[x]<b[y];
 }

 int n;

 void add(int x,int y)
 {
     for(int i=x;i<=n;i+=i&(-i))
         d[i]+=y;
 }

 LL query(int x)
 {
     LL ans=;
     for(int i=x;i>=;i-=i&(-i))
         ans+=d[i];
     return ans;
 }

 int op[Maxn];

 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++) scanf("%d",&a[i]);
     for(int i=;i<=n;i++) scanf("%d",&b[i]);
     for(int i=;i<=n;i++) c[i]=i;
     memset(d,,sizeof(d));
     for(int i=;i<=n;i++) add(i,a[i]);
     for(int i=;i<=n;i++) nt[i]=i+;
     for(int i=;i<=n;i++) lt[i]=i-;
     for(int i=;i<=n;i++) if(a[i]==)
     {
         lt[nt[i]]=lt[i];
         nt[lt[i]]=nt[i];
     }
     sort(c+,c++n,cmp);
     op[]=;
     for(int i=;i<=n;i++)
     {
         if(query(c[i])>=b[c[i]])
         {
             op[++op[]]=c[i];
             int nw=c[i];
             while(nw&&a[nw]<=b[c[i]])
             {
                 add(nw,-a[nw]);
                 b[c[i]]-=a[nw];
                 a[nw]=;
                 lt[nt[nw]]=lt[nw];
                 nt[lt[nw]]=nt[nw];
                 nw=lt[nw];
                 nw--;
             }
             if(b[c[i]])
             {
                 a[nw]-=b[c[i]];
                 add(nw,-b[c[i]]);
             }
         }
     }
     sort(op+,op++op[]);
     printf("%d\n",op[]);
     for(int i=;i<=op[];i++) printf("%d ",op[i]);
     printf("\n");
     return ;
 }

　　方法二：【来者不拒】

　　能选的先选，选了的bi用一个最大堆维护，如果不行就把最大的bi弄出来让现在的满足。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<queue>
 using namespace std;
 #define Maxn 250010
 #define LL long long

 struct node
 {
     int x,id;
     friend bool operator < (node x,node y)
     {
         return x.x<y.x;
     }
 };

 priority_queue<node > q;

 int a[Maxn],b[Maxn];
 bool mark[Maxn];

 int main()
 {
     int n;
     scanf("%d",&n);
     for(int i=;i<=n;i++) scanf("%d",&a[i]);
     for(int i=;i<=n;i++) scanf("%d",&b[i]);
     LL now=;
     int cnt=;
     while(!q.empty()) q.pop();
     memset(mark,,sizeof(mark));
     for(int i=;i<=n;i++)
     {
         now+=a[i];
         if(now>=b[i])
         {
             mark[i]=;
             cnt++;
             q.push((node){b[i],i});
             now-=b[i];
         }
         else if(!q.empty()&&q.top().x>b[i])
         {
             now+=q.top().x-b[i];
             mark[q.top().id]=;
             mark[i]=;
             q.pop();
             q.push((node){b[i],i});
         }
     }
     printf("%d\n",cnt);
     for(int i=;i<=n;i++) if(mark[i])
         printf("%d ",i);printf("\n");
     return ;
 }

2017-01-21 10:47:36