hdu6709 fishing master 2019ccpc网络赛

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=6709

我还记得去年ccpc网络赛第一题也是这种贪心题，放进堆里然后时光倒流。。。

对于煮的久的鱼，我们一定要先钓，因为煮得久，在煮鱼的时候就能钓得更多，出现没鱼可煮然后要登一条鱼钓上来再开煮的情况就可能性更小。

我们再考虑对于每一次煮鱼，如果还剩>=k的时间，那么肯定多钓一条鱼，如果此时剩余煮鱼时间0<t[i]<k，我们就先多钓这条鱼，把k-t[i]放进大根堆里，表示这次多钓一条鱼，多等了这么多时间，用来之后时光倒流，如果t[i]==0，你多钓一条鱼和结束煮鱼之后再去钓鱼是等价的。

如果当前鱼已经钓完了，而当前煮鱼时间内还可以多钓一条鱼，那么就从堆里找一个最大的出来进行时光倒流，把当时钓得那条鱼，放到现在来钓，那次就不用多等了。

#include<bits/stdc++.h>
#define maxl 100010
using namespace std;

int n,k;
int a[maxl],t[maxl];
long long ans;
priority_queue<int> q;

inline bool cmp(const int &x,const int &y)
{
    return x>y;
}

inline void prework()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&t[i]);
    sort(t+1,t+1+n,cmp);
}

inline void mainwork()
{
    while(!q.empty()) q.pop();
    ans=0;
    int cnt=0,sum=0,num;
    for(int i=1;i<=n;i++)
    {
        if(cnt==0)
            ans+=k,cnt++,sum++;
        cnt--;ans+=t[i];
        num=min(n-sum,t[i]/k);
        t[i]-=num*k;
        cnt+=num;sum+=num;
        if(sum<n && t[i]>0)
        {
            cnt++;sum++;
            ans+=k-t[i];
            q.push(k-t[i]);
        }
        else if(sum==n && t[i]>0)
        {
            while(t[i]>=k && !q.empty())
            {
                ans-=q.top();q.pop();
                t[i]-=k;
            }
            if(t[i]>0 && !q.empty())
            {
                if(k-t[i]<q.top())
                {
                    ans-=q.top();q.pop();
                    q.push(k-t[i]);
                    ans+=k-t[i];
                }
            }
        }
    }
}

inline void print()
{
    printf("%lld\n",ans);
}

int main()
{
    int t;
    scanf("%d",&t);
    for(int i=1;i<=t;i++)
    {
        prework();
        mainwork();
        print();
    }
    return 0;
}