dp之购物

#dp
题目大概意思对于t次询问，给你n件商品，由于商场搞活动，你可以一次性挑选k件物品并且只支付其中最贵的那一件，给你p个硬币，问你最多可以买到多少件物品。
t<=1e4, n<=2e5, p<=2e9, k<=n;

样例输入：
8
5 6 2
2 4 3 5 7
5 11 2
2 4 3 5 7
3 2 3
4 2 6
5 2 3
10 1 3 9 2
2 10000 2
10000 10000
2 9999 2
10000 10000
4 6 4
3 2 3 2
5 5 3
1 2 2 1 2
样例输出：
3
4
1
1
2
0
4
5

对于第一个样例来说，如果我们拿一件，那么我们必然是拿最便宜的那件，之花费2块钱，如果我们拿两件，那么我们可以两件分开付钱，就是5块，但是我们可以使用优惠，就只用三块钱就可以了（比如当前位置是pos，那么pos-k+1到pos这段的商品都将因优惠，只需要付款a[pos]元），当我们拿三件物品时，我们可以选择让前面两件使用优惠，也可以计算前pos-k的价格加上pos-k+1到pos这段（只需要支付a[pos]元）…类推…

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+9;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int a[N],dp[N];
int main(void)
{
    
    int t;
    cin>>t;
    while(t--)
    {
        int n,p,k;
        cin>>n>>p>>k;
        for(int i=1;i<=n;i++)
            cin>>a[i];
        sort(a+1,a+1+n);//从小到大排序
        dp[0]=0;
        for(int i=1;i<=n;i++)//状态转移的过程
        {
            dp[i]=dp[i-1]+a[i];
            if(i>=k)
                dp[i]=min(dp[i],dp[i-k]+a[i]);
        }
        for(int i=n;i>=0;i--)
        {
            if(dp[i]<=p)
            {
                cout<<i<<endl;//当取i件物品且此时物品的总花费小于p
                break;
            }
        }
    }
}