题目
给你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 ;
}