Codeforces Round #610 (Div. 2) B2 贪心

题目

给你N件商品和价格和你身上有的钱。
你有两种操作（操作可以执行多次）
一次买一种商品付这个商品的价格
同时买K件商品只需付K件商品中最贵的那件的价格。
求能买的最多的商品。

题解思路

一开始想复杂了，做了很久。
首先排序。
贪心策略
满足K件的时候只需付最贵的价格
并且是K件连着买的东西。
证明
如果隔开来买东西的话。这样需要付更贵的价格，但是得出的件数和钱少一点的相同。
假设他买了N件的话
贪心分配
前N%K件的商品需要付出原价。N%K + K — N%K + 2K — N%K + 3K — N
再加上这些东西的价格。
这样我们可以预处理出前缀和。
K-1件是直接处理。
K的倍数的情况。我们直接N从2K开始加上N-k的价格。往前就能列出前缀和。

AC代码

#include <bits/stdc++.h>
//#include <unordered_map>
//priority_queue
#define PII pair<int,int>
#define ll long long

using namespace std;

const  int  INF =  0x3f3f3f3f;

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int T ;
    cin >> T ; 
    while ( T-- )
    {
        int n , p , k ; 
        cin >> n >> p >> k ;
        vector <int> a(n) ;
        for (int i = 0 ; i < n ; i++ )
            cin >> a[i] ; 
        sort(a.begin(),a.end()) ;
        vector <int> sum(n+1) ;
        for (int i = 0 ; i < n ; i++ )
            sum[i+1] = sum[i] + a[i] ; 
        for (int i = 2*k-1 ; i < n ; i++ )
            a[i] += a[i-k];
        for (int i = 0 ; i < k-1 ; i++ )
            a[i] = 0 ;
        int ans = 0 ; 
        for (int i = 0 ; i < n ; i++ )
        {
            if ( sum[(i+1)%k] + a[i] <= p )
                ans = i+1 ;
        }
        cout << ans << "\n" ; 
    }

    return 0 ;
}