B2. K for the Price of One (Hard Version)

B2. K for the Price of One (Hard Version)
题目链接

题目大意

给你n件商品，你可以一次性买k件商品，付其中最贵的商品一件的价格，或者你单独买一件，付一件的钱，问你最多能买多少件商品。

思路
n件商品价格先排序，从小到大拿，这样是最优的的，当一次性拿k件商品时，肯定在k-1件商品中选价格差最小的物品。在物品拿的物品数没有超过k时，先将前k个物品拿入，当拿的物品超过k时，那么就当前位置的到前k件物品的价格清除只付当前物品的钱，依次往后枚举

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
using namespace std;
#define ll long long
#define sl(n) scanf("%lld",&n)
#define pl(n) printf("%lld",n)
#define sdf(n) scanf("%lf",&n)
#define pdf(n) printf("%.lf",n)
#define pE printf("\n")
#define ull unsigned long long
#define pb push_back
#define pre(n) for(ll i=1;i<=n;i++)
#define rep(n) for(ll i=n;i>=1;i--)
#define pi pair<ll,ll>
#define fi first
#define se second

ll dp[200010];
int main(){
    ll t,n,p,k;
    sl(t);
    while(t--){
        memset(dp,0,sizeof(dp));
       ll  ans=0;
        sl(n),sl(p),sl(k);

        ll a[n+1];

        pre(n)sl(a[i]);

        sort(a+1,a+n+1);

        pre(n){
            if(i<k)dp[i]=dp[i-1]+a[i];
            else {
            dp[i]= a[i]+dp[i-k];
            }
            if(dp[i]<=p)ans=max(i,ans);
        }



        cout<<ans<<endl;
    }
    return 0;
}