CF补题 ---Codeforces Round #610 (Div. 2)B题(hard)

题意：给你n个商品价格，你有p钱，在这里有个k，问你用这些钱最多可以买多少个商品。可以进行两个操作：

1、选择单买一个商品i，花费ai；

2、可以连续买k个商品，花费k个商品中的最大商品价格。

这两个操作都是在p大于你想进行的操作所需要的钱的前提下进行。

思路：当然是进行第二个方案买的商品数量多。在前k个商品中判断前缀和与p的关系，

    #include <bits/stdc++.h>
    using namespace std;
    const int maxn=2e5+100;
    typedef long long ll;
     
    int main()
    {
        int t;
        scanf("%d",&t);
        while(t--)
        {
          int n,k,p;
          scanf("%d%d%d",&n,&p,&k);
          ll a[maxn];
          ll sum[maxn];
          for(int i=0;i<n;i++)
          {
            scanf("%lld",&a[i]);
          }
          sort(a,a+n);
          sum[0]=a[0];
          for(int i=1;i<n;i++)
          {
             sum[i]=sum[i-1]+a[i];
          }
          int num=0;
          int ans=0;
          int flag=0;
          for(int i=0;i<k;i++)
          {
             if(sum[i]>p)
             {
                break;
             }
             num=i;
             int j=i+k-1;
             ll x=0;
             if(i) x+=sum[i-1];
             while(j<n)
             {
                x+=a[j];
                if(x>p)
                {
                   j-=k;
                   break;
                }
                j+=k;
             }
             if(j>=n) j-=k;
             ans=max(max(ans,num+1),j+1);
          }
          if(flag) ans=max(ans,num+1);
          printf("%d\n",ans);
        }
        return 0;
    }