JZOJ 5461. 【NOIP2017提高A组冲刺11.8】购物

5461. 【NOIP2017提高A组冲刺11.8】购物 
(File IO): input:shopping.in output:shopping.out

Time Limits: 1000 ms  Memory Limits: 524288 KB  Detailed Limits  

Goto ProblemSet

Description

X 城的商场中，有着琳琅满目的各种商品。一日，小X 带着小Y 前来购物，小Y 一共看中了n件商品，每一件商品价格为Pi。小X 现在手中共有m个单位的现金，以及k 张优惠券。小X 可以在购买某件商品时，使用至多一张优惠券，若如此做，该商品的价格会下降至Qi。
小X 希望尽可能多地满足小Y 的愿望，所以小X 想要知道他至多能购买多少件商品。

 

Input

第一行包含三个整数n,k,m，表示商品总数，小X 拥有的优惠券与现金总数。
接下来n行每行包含两个整数Pi,Qi。

Output

共一行包含一个整数，表示小X 至多能购买的物品数。

 

Sample Input

4 1 7
3 2
2 2
8 1
4 3

Sample Output

3
样例解释：一种最优的购买方式是购买1,2,3号物品，并用优惠券购买物品3，总共花费为3+2+1=6。
 

 

做法：对Qi 和 Pi 排序， 先最小的K个Qi，然后将这K个Pi - Qi放进一个小根堆，每次比较Pi 和 Qi + 小根堆堆顶的值，更新答案和小根堆。一种具有撤销操作的贪心。

 

代码如下：

 #include <cstdio>
 #include <iostream>
 #include <cstring>
 #include <queue>
 #include <algorithm>
 #define LL long long
 #define N 50007
 using namespace std;
 LL n, m, k, p[N], tot;
 struct arr
 {
     LL pi, qi;
 }q[N];
 priority_queue < LL, vector<LL>, greater<LL> > Q;

 LL cmp(arr x, arr y)
 {
     return x.qi < y.qi;
 }

 int main()
 {
     freopen("shopping.in", "r", stdin);
     freopen("shopping.out", "w", stdout);
     scanf("%lld%lld%lld", &n, &k, &m);
     for (int i = ; i <= n; i++)
         scanf("%lld%lld", &q[i].pi, &q[i].qi);
     sort(q + , q + n + , cmp);
     LL ans = , site = ;
     for (int i = ; i <= k; i++)
     {
         if (m - q[i].qi >= )
         {
             m -= q[i].qi;
             ans++;
             Q.push(q[i].pi - q[i].qi);
             site = i;
         }
         else break;
     }
     if (site < k)
     {
         printf("%lld", ans);
         return ;
     }
     for (int i = site + ; i <= n; i++)
         p[i - site] = q[i].pi;
     tot = n - site;
     sort(p + , p + n - site + );
     LL j = ;
     site++;
     while (m >=  && (j <= tot || site <= n))
     {
         LL now = Q.top();
         LL mi = min(p[j], q[site].qi + now);
         if (m - mi >= )
         {
             if (mi == p[j])
             {
                 m -= p[j];
                 j++;
                 ans++;
                 if (j > tot)    p[j] = 0x7f7f7f7f;
                 if (ans == n)    break;
             }
             else
             {
                 m -= q[site].qi + now;
                 ans++;
                 Q.pop();
                 Q.push(q[site].pi - q[site].qi);
                 site++;
             }
         }
         else break;
     }
     printf("%lld", ans);
 }